Ability to run regression tests using pytest

pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+python_files = test*.py
+python_classes = NoThanks
+filterwarnings = ignore::UserWarning

tests/regression_tests/asymmetric_lattice/test.py

@@ -1,13 +1,11 @@
-#!/usr/bin/env python
-
 import os
-import sys
 import glob
 import hashlib
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import PyAPITestHarness
+
 import openmc
 
+from tests.testing_harness import PyAPITestHarness
+
 
 class AsymmetricLatticeTestHarness(PyAPITestHarness):
     def __init__(self, *args, **kwargs):
@@ -90,6 +88,7 @@ def _get_results(self, hash_output=True):
         return outstr
 
 
-if __name__ == '__main__':
+def test_asymmetric_lattice(request):
     harness = AsymmetricLatticeTestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/cmfd_feed/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import CMFDTestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import CMFDTestHarness
 
-
-if __name__ == '__main__':
+def test_cmfd_feed(request):
     harness = CMFDTestHarness('statepoint.20.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/cmfd_nofeed/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import CMFDTestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import CMFDTestHarness
 
-
-if __name__ == '__main__':
+def test_cmfd_nofeed(request):
     harness = CMFDTestHarness('statepoint.20.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/complex_cell/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_complex_cell(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/confidence_intervals/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_confidence_intervals(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/create_fission_neutrons/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
-
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import PyAPITestHarness
 import openmc
 
+from tests.testing_harness import PyAPITestHarness
+
 
 class CreateFissionNeutronsTestHarness(PyAPITestHarness):
     def _build_inputs(self):
@@ -69,6 +65,7 @@ def _get_results(self):
         return outstr
 
 
-if __name__ == '__main__':
+def test_create_fission_neutrons(request):
     harness = CreateFissionNeutronsTestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/density/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_density(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/diff_tally/test.py

@@ -1,15 +1,12 @@
-#!/usr/bin/env python
-
 import glob
 import os
-import sys
 
 import pandas as pd
-
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import PyAPITestHarness
 import openmc
 
+from tests.testing_harness import PyAPITestHarness
+
+
 class DiffTallyTestHarness(PyAPITestHarness):
     def __init__(self, *args, **kwargs):
         super(DiffTallyTestHarness, self).__init__(*args, **kwargs)
@@ -125,6 +122,7 @@ def _get_results(self):
         return df.to_csv(None, columns=cols, index=False, float_format='%.7e')
 
 
-if __name__ == '__main__':
+def test_diff_tally(request):
     harness = DiffTallyTestHarness('statepoint.3.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/distribmat/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
-
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness, PyAPITestHarness
 import openmc
 
+from tests.testing_harness import TestHarness, PyAPITestHarness
+
 
 class DistribmatTestHarness(PyAPITestHarness):
     def _build_inputs(self):
@@ -103,6 +99,7 @@ def _get_results(self):
         return outstr
 
 
-if __name__ == '__main__':
+def test_distribmat(request):
     harness = DistribmatTestHarness('statepoint.5.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/eigenvalue_genperbatch/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_eigenvalue_genperbatch(request):
     harness = TestHarness('statepoint.7.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/eigenvalue_no_inactive/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_eigenvalue_no_inactive(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/energy_cutoff/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
-
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import PyAPITestHarness
 import openmc
 
+from tests.testing_harness import PyAPITestHarness
+
 
 class EnergyCutoffTestHarness(PyAPITestHarness):
     def _build_inputs(self):
@@ -73,6 +69,7 @@ def _get_results(self):
         return outstr
 
 
-if __name__ == '__main__':
+def test_energy_cutoff(request):
     harness = EnergyCutoffTestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/energy_grid/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
+from tests.testing_harness import TestHarness
 
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
 
-
-if __name__ == '__main__':
+def test_energy_grid(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/energy_laws/test.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-
 """The purpose of this test is to provide coverage of energy distributions that
 are not covered in other tests. It has a single material with the following
 nuclides:
@@ -18,13 +16,10 @@
 
 """
 
-import glob
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
+from tests.testing_harness import TestHarness
 
 
-if __name__ == '__main__':
+def test_energy_laws(request):
     harness = TestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/enrichment/test.py

@@ -1,16 +1,13 @@
-#!/usr/bin/env python
-
 import os
 import sys
 
 import numpy as np
 
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
 from openmc import Material
 from openmc.data import NATURAL_ABUNDANCE, atomic_mass
 
 
-if __name__ == '__main__':
+def test_enrichment():
     # This test doesn't require an OpenMC run.  We just need to make sure the
     # element.expand() method expands Uranium to the proper enrichment.
 

tests/regression_tests/entropy/test.py

@@ -1,12 +1,10 @@
-#!/usr/bin/env python
-
 import glob
 import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import TestHarness
+
 from openmc import StatePoint
 
+from tests.testing_harness import TestHarness
+
 
 class EntropyTestHarness(TestHarness):
     def _get_results(self):
@@ -26,6 +24,7 @@ def _get_results(self):
         return outstr
 
 
-if __name__ == '__main__':
+def test_entropy(request):
     harness = EntropyTestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/filter_distribcell/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
-
 import glob
-import hashlib
 import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import *
+
+from tests.testing_harness import *
 
 
 class DistribcellTestHarness(TestHarness):
@@ -72,6 +68,7 @@ def _test_output_created(self, tallies_out_present):
                  'Tally output file does not exist.'
 
 
-if __name__ == '__main__':
+def test_filter_distribcell(request):
     harness = DistribcellTestHarness()
+    harness.request = request
     harness.main()

tests/regression_tests/filter_energyfun/test.py

@@ -1,12 +1,7 @@
-#!/usr/bin/env python
-
-import os
-import sys
-import glob
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import PyAPITestHarness
 import openmc
 
+from tests.testing_harness import PyAPITestHarness
+
 
 class FilterEnergyFunHarness(PyAPITestHarness):
     def __init__(self, *args, **kwargs):
@@ -38,8 +33,7 @@ def __init__(self, *args, **kwargs):
 
     def _get_results(self):
         # Read the statepoint file.
-        statepoint = glob.glob(os.path.join(os.getcwd(), self._sp_name))[0]
-        sp = openmc.StatePoint(statepoint)
+        sp = openmc.StatePoint(self._sp_name)
 
         # Use tally arithmetic to compute the branching ratio.
         br_tally = sp.tallies[2] / sp.tallies[1]
@@ -48,6 +42,7 @@ def _get_results(self):
         return br_tally.get_pandas_dataframe().to_string() + '\n'
 
 
-if __name__ == '__main__':
+def test_filter_energyfun(request):
     harness = FilterEnergyFunHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()

tests/regression_tests/filter_mesh/test.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
-
-import os
-import sys
-sys.path.insert(0, os.path.join(os.pardir, os.pardir))
-from testing_harness import HashedPyAPITestHarness
 import openmc
 
+from tests.testing_harness import HashedPyAPITestHarness
+
 
 class FilterMeshTestHarness(HashedPyAPITestHarness):
     def __init__(self, *args, **kwargs):
@@ -67,6 +63,7 @@ def __init__(self, *args, **kwargs):
         self._model.tallies.append(tally)
 
 
-if __name__ == '__main__':
+def test_filter_mesh(request):
     harness = FilterMeshTestHarness('statepoint.10.h5')
+    harness.request = request
     harness.main()