crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-26 13:15:39 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix openmc-track-to-vtk and track test

Browse source
This commit is contained in:
Paul Romano 2017-04-07 17:26:58 -05:00
parent 44949016b5
commit 3e7bca49a8
2 changed files with 8 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
scripts/openmc-track-to-vtk
View file

@ -42,8 +42,8 @@ def main():
for fname in args.input:
# Write coordinate values to points array.
track = h5py.File(fname)
n_particles = track['n_particles'].value
n_coords = track['n_coords']
n_particles = track.attrs['n_particles']
n_coords = track.attrs['n_coords']
coords = []
for i in range(n_particles):
coords.append(track['coordinates_' + str(i + 1)].value)

18
tests/test_track_output/test_track_output.py
View file

@ -14,32 +14,26 @@ class TrackTestHarness(TestHarness):
"""Make sure statepoint.* and track* have been created."""
TestHarness._test_output_created(self)
outputs = [glob.glob(''.join((os.getcwd(), '/track_1_1_1.*')))]
outputs.append(glob.glob(''.join((os.getcwd(), '/track_1_1_2.*'))))
for files in outputs:
assert len(files) == 1, 'Multiple or no track files detected.'
assert files[0].endswith('h5'),\
'Track files are not HDF5 files'
outputs = glob.glob('track_1_1_*.h5')
assert len(outputs) == 2, 'Expected two track files.'
def _get_results(self):
"""Digest info in the statepoint and return as a string."""
# Run the track-to-vtk conversion script.
call(['../../scripts/openmc-track-to-vtk', '-o', 'poly'] +
glob.glob(''.join((os.getcwd(), '/track*'))))
glob.glob('track_1_1_*.h5'))
# Make sure the vtk file was created then return it's contents.
poly = os.path.join(os.getcwd(), 'poly.pvtp')
assert os.path.isfile(poly), 'poly.pvtp file not found.'
assert os.path.isfile('poly.pvtp'), 'poly.pvtp file not found.'
with open(poly) as fin:
with open('poly.pvtp', 'r') as fin:
outstr = fin.read()
return outstr
def _cleanup(self):
TestHarness._cleanup(self)
output = glob.glob(os.path.join(os.getcwd(), 'track*'))
output += glob.glob(os.path.join(os.getcwd(), 'poly*'))
output = glob.glob('track*') + glob.glob('poly*')
for f in output:
if os.path.exists(f):
os.remove(f)