Merge pull request #452 from smharper/cleanup_sourcesep

Make test cleanup source file

tests/test_entropy/test_entropy.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_filter_distribcell/test_filter_distribcell.py

@@ -1,6 +1,8 @@
 #!/usr/bin/env python
 
+import glob
 import hashlib
+import os
 import sys
 
 sys.path.insert(0, '..')

tests/test_fixed_source/test_fixed_source.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_output/test_output.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_source_file/test_source_file.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_sourcepoint_batch/test_sourcepoint_batch.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_sourcepoint_interval/test_sourcepoint_interval.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_sourcepoint_latest/test_sourcepoint_latest.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_statepoint_restart/test_statepoint_restart.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 

tests/test_statepoint_sourcesep/test_statepoint_sourcesep.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import sys
+
 sys.path.insert(0, '..')
 from testing_harness import *
 
@@ -15,6 +18,13 @@ class SourcepointTestHarness(TestHarness):
         assert source[0].endswith('h5'), \
              'Source file is not a HDF5 file.'
 
+    def _cleanup(self):
+        TestHarness._cleanup(self)
+        output = glob.glob(os.path.join(os.getcwd(), 'source.*'))
+        for f in output:
+            if os.path.exists(f):
+                os.remove(f)
+
 
 if __name__ == '__main__':
     harness = SourcepointTestHarness('statepoint.10.*')

tests/test_track_output/test_track_output.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 
+import glob
+import os
 import shutil
 import sys