From 228c9eaba825e56d570ec6ce7076933d913f2974 Mon Sep 17 00:00:00 2001 From: Mark Driver <mdd31@alumni.cam.ac.uk> Date: Tue, 4 Feb 2020 18:15:48 +0000 Subject: [PATCH] update file clean up. --- .../test/runnerstest/phasexmlcreatorrunnertest.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/phasecalculator/test/runnerstest/phasexmlcreatorrunnertest.py b/phasecalculator/test/runnerstest/phasexmlcreatorrunnertest.py index 612b2fa..01cb9e2 100755 --- a/phasecalculator/test/runnerstest/phasexmlcreatorrunnertest.py +++ b/phasecalculator/test/runnerstest/phasexmlcreatorrunnertest.py @@ -75,6 +75,15 @@ class PhaseXMLCreatorRunnerTestCase(unittest.TestCase): None. """ + for filename in os.listdir(self.runtime_inf.scratch_dir): + file_path = os.path.join(self.runtime_inf.scratch_dir, filename) + try: + if os.path.isfile(file_path) or os.path.islink(file_path): + os.unlink(file_path) + elif os.path.isdir(file_path): + shutil.rmtree(file_path) + except Exception as e: + LOGGER.error('Failed to delete %s. Reason: %s', file_path, e) def test_create_phase_and_solvent_files(self): """Test -- GitLab