added test for restarting with a sourcepoint

tests/test_sourcepoint_restart/geometry.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0"?>
+<geometry>
+
+  <!-- Sphere with radius 10 -->
+  <surface id="1" type="sphere" coeffs="0 0 0 10" boundary="vacuum"/>
+  <cell id="1" material="1" surfaces="-1" />
+    
+</geometry>

tests/test_sourcepoint_restart/materials.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0"?>
+<materials>
+
+  <material id="1">
+    <density value="4.5" units="g/cc" />
+    <nuclide name="U-235" xs="70c" ao="1.0" />
+  </material>
+
+</materials>

tests/test_sourcepoint_restart/results.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python
+
+import sys
+import numpy as np
+
+# import statepoint
+sys.path.append('../../src/utils')
+import statepoint
+
+# read in statepoint file
+if len(sys.argv) > 1:
+    sp = statepoint.StatePoint(sys.argv[1])
+else:
+    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:12.6E} {1:12.6E}\n".format(sp.k_combined[0], sp.k_combined[1])
+
+# write out tally results
+outstr += 'tally 1:\n'
+for item in results1:
+  outstr += "{0:12.6E}\n".format(item)
+outstr += 'tally 2:\n'
+for item in results2:
+  outstr += "{0:12.6E}\n".format(item)
+
+# write results to file
+with open('results_test.dat','w') as fh:
+    fh.write(outstr)

tests/test_sourcepoint_restart/settings.xml

@@ -0,0 +1,19 @@
+<?xml version="1.0"?>
+<settings>
+
+  <state_point batches="7" />
+  <source_point source_separate="true" />
+
+  <eigenvalue>
+    <batches>10</batches>
+    <inactive>5</inactive>
+    <particles>1000</particles>
+  </eigenvalue>
+
+  <source>
+    <space type="box">
+      <parameters>-4 -4 -4  4  4  4</parameters>
+    </space>
+  </source>
+
+</settings>

tests/test_sourcepoint_restart/tallies.xml

@@ -0,0 +1,23 @@
+<?xml version="1.0"?>
+<tallies>
+
+  <mesh id="1">
+    <type>rectangular</type>
+    <dimension>5 3 4</dimension>
+    <lower_left>-10. -5. 0.</lower_left>
+    <upper_right>10. 4. 9.</upper_right>
+  </mesh>
+
+  <tally id="10">
+    <filter type="mesh" bins="1" />
+    <filter type="energy" bins="0.0 5.0 10.0" />
+    <filter type="energyout" bins="0.0 5.0 10.0" />
+    <scores>scatter-P3 nu-fission</scores>
+  </tally>
+
+  <tally id="5">
+    <filter type="cell" bins="1" />
+    <scores>fission absorption total flux</scores>
+  </tally>
+
+</tallies>

tests/test_sourcepoint_restart/test_sourcepoint_restart.py

@@ -0,0 +1,133 @@
+#!/usr/bin/env python
+
+import os
+from subprocess import Popen, STDOUT, PIPE, call
+import filecmp
+from nose_mpi import NoseMPI
+import glob
+
+pwd = os.path.dirname(__file__)
+
+def setup(): 
+    os.putenv('PWD', pwd)
+    os.chdir(pwd)
+
+def test_run():
+    openmc_path = pwd + '/../../src/openmc'
+    if int(NoseMPI.mpi_np) > 0:
+        proc = Popen([NoseMPI.mpi_exec, '-np', NoseMPI.mpi_np, openmc_path],
+               stderr=STDOUT, stdout=PIPE)
+    else:
+        proc = Popen([openmc_path], stderr=STDOUT, stdout=PIPE)
+    print(proc.communicate()[0])
+    returncode = proc.returncode
+    assert returncode == 0
+
+def test_created_statepoint():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    assert len(statepoint) == 1
+    assert statepoint[0].endswith('binary') or statepoint[0].endswith('h5')
+    sourcepoint = glob.glob(pwd + '/source.7.*')
+    assert len(sourcepoint) == 1
+    assert sourcepoint[0].endswith('binary') or sourcepoint[0].endswith('h5')
+
+def test_results():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    call(['python', 'results.py', statepoint[0]])
+    compare = filecmp.cmp('results_test.dat', 'results_true.dat')
+    if not compare:
+      os.rename('results_test.dat', 'results_error.dat')
+    assert compare
+
+def test_restart_form1():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    openmc_path = pwd + '/../../src/openmc'
+    if int(NoseMPI.mpi_np) > 0:
+        proc = Popen([NoseMPI.mpi_exec, '-np', NoseMPI.mpi_np, openmc_path, 
+                      '-r', statepoint[0]],
+               stderr=STDOUT, stdout=PIPE)
+    else:
+        proc = Popen([openmc_path, '-r', statepoint[0]], stderr=STDOUT, stdout=PIPE)
+    print(proc.communicate()[0])
+    returncode = proc.returncode
+    assert returncode == 0
+
+def test_created_statepoint_form1():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    assert len(statepoint) == 1
+    assert statepoint[0].endswith('binary') or statepoint[0].endswith('h5')
+    sourcepoint = glob.glob(pwd + '/source.7.*')
+    assert len(sourcepoint) == 1
+    assert sourcepoint[0].endswith('binary') or sourcepoint[0].endswith('h5')
+
+def test_results_form1():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    call(['python', 'results.py', statepoint[0]])
+    compare = filecmp.cmp('results_test.dat', 'results_true.dat')
+    if not compare:
+      os.rename('results_test.dat', 'results_error.dat')
+    assert compare
+
+def test_restart_form2():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    sourcepoint = glob.glob(pwd + '/source.7.*')
+    openmc_path = pwd + '/../../src/openmc'
+    if int(NoseMPI.mpi_np) > 0:
+        proc = Popen([NoseMPI.mpi_exec, '-np', NoseMPI.mpi_np, openmc_path, 
+                      '--restart', statepoint[0], source[0]],
+               stderr=STDOUT, stdout=PIPE)
+    else:
+        proc = Popen([openmc_path, '--restart', statepoint[0]], 
+                     stderr=STDOUT, stdout=PIPE)
+    print(proc.communicate()[0])
+    returncode = proc.returncode
+    assert returncode == 0
+
+def test_created_statepoint_form2():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    assert len(statepoint) == 1
+    assert statepoint[0].endswith('binary') or statepoint[0].endswith('h5')
+    sourcepoint = glob.glob(pwd + '/source.7.*')
+    assert len(sourcepoint) == 1
+    assert sourcepoint[0].endswith('binary') or sourcepoint[0].endswith('h5')
+
+def test_results_form2():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    call(['python', 'results.py', statepoint[0]])
+    compare = filecmp.cmp('results_test.dat', 'results_true.dat')
+    if not compare:
+      os.rename('results_test.dat', 'results_error.dat')
+    assert compare
+
+def test_restart_serial():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    openmc_path = pwd + '/../../src/openmc'
+    proc = Popen([openmc_path, '--restart', statepoint[0]], 
+                 stderr=STDOUT, stdout=PIPE)
+    print(proc.communicate()[0])
+    returncode = proc.returncode
+    assert returncode == 0
+
+def test_created_statepoint_serial():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    assert len(statepoint) == 1
+    assert statepoint[0].endswith('binary') or statepoint[0].endswith('h5')
+    sourcepoint = glob.glob(pwd + '/source.7.*')
+    assert len(sourcepoint) == 1
+    assert sourcepoint[0].endswith('binary') or sourcepoint[0].endswith('h5')
+
+def test_results_serial():
+    statepoint = glob.glob(pwd + '/statepoint.7.*')
+    call(['python', 'results.py', statepoint[0]])
+    compare = filecmp.cmp('results_test.dat', 'results_true.dat')
+    if not compare:
+      os.rename('results_test.dat', 'results_error.dat')
+    assert compare
+
+def teardown():
+    output = glob.glob(pwd + '/statepoint.7.*')
+    output += glob.glob(pwd + '/sourcepoint.7.*')
+    output.append(pwd + '/results_test.dat')
+    for f in output:
+        if os.path.exists(f):
+            os.remove(f)