From 48a95a5b8d195174e16dc465c5f81a7d8441a47e Mon Sep 17 00:00:00 2001
From: Qingming He <906459647@qq.com>
Date: Tue, 11 Oct 2016 14:36:10 -0400
Subject: [PATCH] add an unit test for energy cutoff

---
 tests/test_energy_cutoff/inputs_true.dat      |  1 +
 tests/test_energy_cutoff/results_true.dat     |  3 +
 .../test_energy_cutoff/test_energy_cutoff.py  | 84 +++++++++++++++++++
 3 files changed, 88 insertions(+)
 create mode 100644 tests/test_energy_cutoff/inputs_true.dat
 create mode 100644 tests/test_energy_cutoff/results_true.dat
 create mode 100755 tests/test_energy_cutoff/test_energy_cutoff.py

diff --git a/tests/test_energy_cutoff/inputs_true.dat b/tests/test_energy_cutoff/inputs_true.dat
new file mode 100644
index 0000000000..0630d2acc2
--- /dev/null
+++ b/tests/test_energy_cutoff/inputs_true.dat
@@ -0,0 +1 @@
+4dc6a7b131f6757ecc9b06e93d425712f0813c546ed9eaa0aafa4990383b2f3a5b742a5a39ef5f27300d2c861b344e78de2d2c569b9439e7893dc06b03bf3b02
\ No newline at end of file
diff --git a/tests/test_energy_cutoff/results_true.dat b/tests/test_energy_cutoff/results_true.dat
new file mode 100644
index 0000000000..5ad463b774
--- /dev/null
+++ b/tests/test_energy_cutoff/results_true.dat
@@ -0,0 +1,3 @@
+tally 1:
+sum = 0.000000E+00
+sum_sq = 0.000000E+00
diff --git a/tests/test_energy_cutoff/test_energy_cutoff.py b/tests/test_energy_cutoff/test_energy_cutoff.py
new file mode 100755
index 0000000000..6467a59255
--- /dev/null
+++ b/tests/test_energy_cutoff/test_energy_cutoff.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+
+import os
+import sys
+sys.path.insert(0, os.pardir)
+from testing_harness import PyAPITestHarness
+import openmc
+
+
+class EnergyCutoffTestHarness(PyAPITestHarness):
+    def _build_inputs(self):
+        # Set energy cutoff
+        energy_cutoff = 4e-6
+
+        # Material is composed of H-1
+        mat = openmc.Material(material_id=1, name='mat')
+        mat.set_density('atom/b-cm', 0.069335)
+        mat.add_nuclide('H1', 40.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.cutoff = {'energy': energy_cutoff}
+        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()
+
+        # Tally flux under energy cutoff
+        tallies = openmc.Tallies()
+        tally = openmc.Tally(1)
+        tally.scores = ['flux']
+        energy_filter = openmc.filter.EnergyFilter((0.0, energy_cutoff))
+        tally.filters = [energy_filter]
+        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 {0}:\n'.format(t.id)
+        outstr += 'sum = {0:12.6E}\n'.format(t.sum[0, 0, 0])
+        outstr += 'sum_sq = {0:12.6E}\n'.format(t.sum_sq[0, 0, 0])
+
+        return outstr
+
+    def _cleanup(self):
+        super(EnergyCutoffTestHarness, self)._cleanup()
+        f = os.path.join(os.getcwd(), 'tallies.xml')
+        if os.path.exists(f):
+            os.remove(f)
+
+
+if __name__ == '__main__':
+    harness = EnergyCutoffTestHarness('statepoint.10.h5', True)
+    harness.main()