From b9379e3c8667d062ec6511ad07f2525ea0b2f5ef Mon Sep 17 00:00:00 2001 From: Sterling Harper Date: Mon, 14 Sep 2015 21:05:58 -0400 Subject: [PATCH] Make test cleanup source file --- .../test_statepoint_sourcesep/test_statepoint_sourcesep.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_statepoint_sourcesep/test_statepoint_sourcesep.py b/tests/test_statepoint_sourcesep/test_statepoint_sourcesep.py index acbb0180bf..157210de6f 100644 --- a/tests/test_statepoint_sourcesep/test_statepoint_sourcesep.py +++ b/tests/test_statepoint_sourcesep/test_statepoint_sourcesep.py @@ -15,6 +15,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.*')