Running some existing regression tests using single XML file

paulromano · Dec 24, 2022 · Nov 3, 2022 · Nov 4, 2022 · Nov 4, 2022 · Nov 4, 2022
commit 662d83a7dd72240b4da8594319bab416977ec71b
diff --git a/tests/regression_tests/adj_cell_rotation/__init__.py b/tests/regression_tests/adj_cell_rotation/__init__.py
@@ -0,0 +1,2 @@
+
+from .test import model
diff --git a/tests/regression_tests/energy_laws/__init__.py b/tests/regression_tests/energy_laws/__init__.py
@@ -0,0 +1 @@
+from .test import model
diff --git a/tests/regression_tests/lattice_multiple/__init__.py b/tests/regression_tests/lattice_multiple/__init__.py
@@ -0,0 +1 @@
+from .test import model
diff --git a/tests/regression_tests/model_xml/__init__.py b/tests/regression_tests/model_xml/__init__.py
diff --git a/tests/regression_tests/model_xml/inputs_true.dat b/tests/regression_tests/model_xml/inputs_true.dat
@@ -0,0 +1,38 @@
+<?xml version='1.0' encoding='utf-8'?>
+<model>
+<materials>
+  <material depletable="true" id="1">
+    <density units="g/cc" value="10.0" />
+    <nuclide ao="1.0" name="U235" />
+  </material>
+  <material id="2">
+    <density units="g/cc" value="0.1" />
+    <nuclide ao="0.1" name="H1" />
+  </material>
+</materials>
+<geometry>
+  <cell id="1" material="1" region="-1" universe="1" />
+  <cell id="2" material="2" region="1" universe="1" />
+  <cell id="3" fill="1" region="2 -3 4 -5 6 -8" rotation="10 20 30" universe="2" />
+  <cell id="4" fill="1" region="2 -3 4 -5 8 -7" translation="0 0 15" universe="2" />
+  <surface id="1" type="sphere" coeffs="1.0 0.0 0.0 5.0" />
+  <surface id="2" name="minimum x" type="x-plane" boundary="vacuum" coeffs="-7.5" />
+  <surface id="3" name="maximum x" type="x-plane" boundary="vacuum" coeffs="7.5" />
+  <surface id="4" name="minimum y" type="y-plane" boundary="vacuum" coeffs="-7.5" />
+  <surface id="5" name="maximum y" type="y-plane" boundary="vacuum" coeffs="7.5" />
+  <surface id="6" type="z-plane" boundary="vacuum" coeffs="-7.5" />
+  <surface id="7" type="z-plane" boundary="vacuum" coeffs="22.5" />
+  <surface id="8" type="z-plane" coeffs="7.5" />
+</geometry>
+<settings>
+  <run_mode>eigenvalue</run_mode>
+  <particles>10000</particles>
+  <batches>10</batches>
+  <inactive>5</inactive>
+  <source strength="1.0">
+    <space type="box">
+      <parameters>-4.0 -4.0 -4.0 4.0 4.0 4.0</parameters>
+    </space>
+  </source>
+</settings>
+</model>
diff --git a/tests/regression_tests/model_xml/test.py b/tests/regression_tests/model_xml/test.py
@@ -0,0 +1,76 @@
+from difflib import unified_diff
+import filecmp
+import os
+
+import openmc
+import pytest
+
+from tests.testing_harness import PyAPITestHarness, colorize
+
+# use a few models from other tests to make sure the same results are
+# produced when using a single model.xml file as input
+from ..adj_cell_rotation import model as adj_cell_rotation_model
+from ..lattice_multiple import model as lattice_mult_model
+from ..energy_laws import model as energy_laws_model
+from ..photon_production import model as photon_prod_model
+
+
+class ModelXMLTestHarness(PyAPITestHarness):
+    """Accept a results file to check against and assume inputs_true is the contents of a model.xml file.
+    """
+    def __init__(self, model=None, inputs_true=None, results_true=None):
+        statepoint_name = f'statepoint.{model.settings.batches}.h5'
+        super().__init__(statepoint_name, model, inputs_true)
+
+        self.results_true = 'results_true.dat' if results_true is None else results_true
+
+    def _build_inputs(self):
+        self._model.export_to_xml(separate_xmls=False)
+
+    def _get_inputs(self):
+        return open('model.xml').read()
+
+    def _compare_inputs(self):
+        """Skip input comparisons for now
+        """
+        pass
+
+    def _compare_results(self):
+        """Make sure the current results agree with the reference."""
+        compare = filecmp.cmp('results_test.dat', self.results_true)
+        if not compare:
+            expected = open(self.results_true).readlines()
+            actual = open('results_test.dat').readlines()
+            diff = unified_diff(expected, actual, self.results_true,
+                                'results_test.dat')
+            print('Result differences:')
+            print(''.join(colorize(diff)))
+            os.rename('results_test.dat', 'results_error.dat')
+        assert compare, 'Results do not agree'
+
+    def _cleanup(self):
+        super()._cleanup()
+        if os.path.exists('model.xml'):
+            os.remove('model.xml')
+
+
+models = [
+'adj_cell_rotation_model',
+'lattice_mult_model',
+'energy_laws_model',
+'photon_prod_model'
+]
+paths = [
+'../adj_cell_rotation',
+'../lattice_multiple',
+'../energy_laws',
+'../photon_production'
+]
+
+
+@pytest.mark.parametrize("model, path", zip(models, paths))
+def test_model_xml(model, path, request):
+    inputs = path + "/inputs_true.dat"
+    results = path + "/results_true.dat"
+    harness = ModelXMLTestHarness(request.getfixturevalue(model), inputs, results)
+    harness.main()
diff --git a/tests/regression_tests/photon_production/__init__.py b/tests/regression_tests/photon_production/__init__.py
@@ -0,0 +1 @@
+from .test import model