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