Make test cleanup source file

This commit is contained in:
Sterling Harper 2015-09-14 21:05:58 -04:00
parent f8bfa401cb
commit b9379e3c86

View file

@ -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.*')