diff --git a/tests/test_statepoint_batch/results.py b/tests/test_statepoint_batch/results.py new file mode 100644 index 0000000000..6e18864fa2 --- /dev/null +++ b/tests/test_statepoint_batch/results.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python + +import sys + +# import statepoint +sys.path.append('../../src/utils') +import statepoint + +# read in statepoint file +sp = statepoint.StatePoint('statepoint.9.binary') +sp.read_results() + +# set up output string +outstr = '' + +# write out k-combined +outstr += 'k-combined:\n' +outstr += "{0:10.8f} {1:10.8f}".format(sp.k_combined[0], sp.k_combined[1]) + +# write results to file +with open('results_test.dat','w') as fh: + fh.write(outstr) diff --git a/tests/test_statepoint_batch/results_true.dat b/tests/test_statepoint_batch/results_true.dat new file mode 100644 index 0000000000..b9482653df --- /dev/null +++ b/tests/test_statepoint_batch/results_true.dat @@ -0,0 +1,2 @@ +k-combined: +0.30308270 0.00706340 \ No newline at end of file diff --git a/tests/test_statepoint_batch/test_statepoint_batch.py b/tests/test_statepoint_batch/test_statepoint_batch.py index 2c77b48d12..c488f3bd45 100644 --- a/tests/test_statepoint_batch/test_statepoint_batch.py +++ b/tests/test_statepoint_batch/test_statepoint_batch.py @@ -2,6 +2,7 @@ import os from subprocess import Popen, STDOUT, PIPE +import filecmp pwd = os.path.dirname(__file__) @@ -20,10 +21,18 @@ def test_statepoints_exist(): assert os.path.exists(pwd + '/statepoint.6.binary') assert os.path.exists(pwd + '/statepoint.9.binary') +def test_results(): + os.system('python results.py') + compare = filecmp.cmp('results_test.dat', 'results_true.dat') + if not compare: + os.system('cp results_test.dat results_error.dat') + assert compare + def teardown(): output = [pwd + i for i in ['/statepoint.3.binary', '/statepoint.6.binary', - '/statepoint.9.binary']] + '/statepoint.9.binary', + '/results_test.dat']] for f in output: if os.path.exists(f): os.remove(f)