mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-28 22:26:08 -04:00
70 lines
2.5 KiB
Python
Executable file
70 lines
2.5 KiB
Python
Executable file
import openmc
|
|
|
|
from tests.testing_harness import PyAPITestHarness
|
|
|
|
|
|
class CreateFissionNeutronsTestHarness(PyAPITestHarness):
|
|
def _build_inputs(self):
|
|
# Material is composed of H-1 and U-235
|
|
mat = openmc.Material(material_id=1, name='mat')
|
|
mat.set_density('atom/b-cm', 0.069335)
|
|
mat.add_nuclide('H1', 40.0)
|
|
mat.add_nuclide('U235', 1.0)
|
|
materials_file = openmc.Materials([mat])
|
|
materials_file.export_to_xml()
|
|
|
|
# Cell is box with reflective boundary
|
|
x1 = openmc.XPlane(surface_id=1, x0=-1)
|
|
x2 = openmc.XPlane(surface_id=2, x0=1)
|
|
y1 = openmc.YPlane(surface_id=3, y0=-1)
|
|
y2 = openmc.YPlane(surface_id=4, y0=1)
|
|
z1 = openmc.ZPlane(surface_id=5, z0=-1)
|
|
z2 = openmc.ZPlane(surface_id=6, z0=1)
|
|
for surface in [x1, x2, y1, y2, z1, z2]:
|
|
surface.boundary_type = 'reflective'
|
|
box = openmc.Cell(cell_id=1, name='box')
|
|
box.region = +x1 & -x2 & +y1 & -y2 & +z1 & -z2
|
|
box.fill = mat
|
|
root = openmc.Universe(universe_id=0, name='root universe')
|
|
root.add_cell(box)
|
|
geometry = openmc.Geometry(root)
|
|
geometry.export_to_xml()
|
|
|
|
# Set the running parameters
|
|
settings_file = openmc.Settings()
|
|
settings_file.run_mode = 'fixed source'
|
|
settings_file.batches = 10
|
|
settings_file.particles = 100
|
|
settings_file.create_fission_neutrons = False
|
|
bounds = [-1, -1, -1, 1, 1, 1]
|
|
uniform_dist = openmc.stats.Box(bounds[:3], bounds[3:])
|
|
watt_dist = openmc.stats.Watt()
|
|
settings_file.source = openmc.source.Source(space=uniform_dist,
|
|
energy=watt_dist)
|
|
settings_file.export_to_xml()
|
|
|
|
# Create tallies
|
|
tallies = openmc.Tallies()
|
|
tally = openmc.Tally(1)
|
|
tally.scores = ['flux']
|
|
tallies.append(tally)
|
|
tallies.export_to_xml()
|
|
|
|
def _get_results(self):
|
|
"""Digest info in the statepoint and return as a string."""
|
|
# Read the statepoint file.
|
|
sp = openmc.StatePoint(self._sp_name)
|
|
|
|
# Write out tally data.
|
|
outstr = ''
|
|
t = sp.get_tally()
|
|
outstr += 'tally {}:\n'.format(t.id)
|
|
outstr += 'sum = {:12.6E}\n'.format(t.sum[0, 0, 0])
|
|
outstr += 'sum_sq = {:12.6E}\n'.format(t.sum_sq[0, 0, 0])
|
|
|
|
return outstr
|
|
|
|
|
|
def test_create_fission_neutrons():
|
|
harness = CreateFissionNeutronsTestHarness('statepoint.10.h5')
|
|
harness.main()
|