Fix import for statepoint for test_union_energy_grids

This commit is contained in:
Paul Romano 2015-04-16 14:35:49 -05:00
parent 4bf5f3918e
commit 52a4a10d54

View file

@ -2,20 +2,20 @@
import sys
# import statepoint
sys.path.insert(0, '../../src/utils')
import statepoint
from openmc.statepoint import StatePoint
# read in statepoint file
if len(sys.argv) > 1:
sp = statepoint.StatePoint(sys.argv[1])
print(sys.argv)
sp = StatePoint(sys.argv[1])
else:
sp = statepoint.StatePoint('statepoint.10.binary')
sp = StatePoint('statepoint.10.binary')
sp.read_results()
# 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])