updated statepoint_restart test

tests/test_statepoint_restart/results.py

@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+
+import sys
+import numpy as np
+
+# import statepoint
+sys.path.append('../../src/utils')
+import statepoint
+
+# read in statepoint file
+sp = statepoint.StatePoint('statepoint.7.binary')
+sp.read_results()
+
+# extract tally results and convert to vector
+results1 = sp.tallies[0].results
+shape1 = results1.shape
+size1 = (np.product(shape1))
+results1 = np.reshape(results1, size1)
+results2 = sp.tallies[1].results
+shape2 = results2.shape
+size2 = (np.product(shape2))
+results2 = np.reshape(results2, size2)
+
+# set up output string
+outstr = ''
+ 
+# write out k-combined
+outstr += 'k-combined:\n'
+outstr += "{0:10.8f} {1:10.8f}\n".format(sp.k_combined[0], sp.k_combined[1])
+
+# write out tally results
+outstr += 'tally 1:\n'
+for item in results1:
+  outstr += "{0:10.8f}\n".format(item)
+outstr += 'tally 2:\n'
+for item in results2:
+  outstr += "{0:10.8f}\n".format(item)
+
+# write results to file
+with open('results_test.dat','w') as fh:
+    fh.write(outstr)

tests/test_statepoint_restart/test_statepoint_restart.py

@@ -2,6 +2,7 @@
 
 import os
 from subprocess import Popen, STDOUT, PIPE
+import filecmp
 
 pwd = os.path.dirname(__file__)
 
@@ -16,6 +17,13 @@ def test_run():
     assert returncode == 0
     assert os.path.exists(pwd + '/statepoint.7.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 test_restart_form1():
     proc = Popen([pwd + '/../../src/openmc', '-r',
                   pwd + '/statepoint.7.binary'],
@@ -24,6 +32,13 @@ def test_restart_form1():
     print(proc.communicate()[0])
     assert returncode == 0
 
+def test_results_form1():
+    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 test_restart_form2():
     proc = Popen([pwd + '/../../src/openmc', '--restart',
                   pwd + '/statepoint.7.binary'],
@@ -32,8 +47,15 @@ def test_restart_form2():
     print(proc.communicate()[0])
     assert returncode == 0
 
+def test_results_form2():
+    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 + '/statepoint.7.binary']
+    output = [pwd + '/statepoint.7.binary', pwd + '/results_test.dat']
     for f in output:
         if os.path.exists(f):
             os.remove(f)