crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 05:35:49 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding initial test for exporting model.xmls. Correcting material maps

Browse source
This commit is contained in:
Patrick Shriwise 2022-11-04 22:45:37 -05:00
parent 611475a835
commit 021bb280c9
2 changed files with 23 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
openmc/model/model.py
View file

@ -230,8 +230,9 @@ class Model:
model.settings = openmc.Settings.from_xml_element(root.find('settings'))
model.materials = openmc.Materials.from_xml_element(root.find('materials'))
materials = {str(m.id): m for m in model.materials}
model.geometry = \
openmc.Geometry.from_xml_element(root.find('geometry'), model.materials)
openmc.Geometry.from_xml_element(root.find('geometry'), materials)
if root.find('tallies'):
model.tallies = openmc.Tallies.from_xml_element(root.find('tallies'))

21
tests/unit_tests/test_model.py
View file

@ -7,6 +7,8 @@ import pytest
import openmc
import openmc.lib
from .. import cdtemp
@pytest.fixture(scope='function')
def pin_model_attributes():
@ -529,3 +531,22 @@ def test_calc_volumes(run_in_tmpdir, pin_model_attributes, mpi_intracomm):
assert openmc.lib.materials[3].volume == mats[2].volume
test_model.finalize_lib()
def test_model_xml():
with cdtemp():
# load a model from examples
pwr_model = openmc.examples.pwr_core()
# export to separate XMLs manually
pwr_model.settings.export_to_xml('settings_ref.xml')
pwr_model.materials.export_to_xml('materials_ref.xml')
pwr_model.geometry.export_to_xml('geometry_ref.xml')
# now write and read a model.xml file
pwr_model.export_to_xml(separate_xmls=False)
new_model = openmc.Model.from_xml(separate_xmls=False)
# make sure we can also export this again to separate
# XML files
new_model.export_to_xml(separate_xmls=True)