updated type check for energy

fusion-energy · Mar 30, 2023 · ed2b47a · ed2b47a
1 parent 0523319
commit ed2b47a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_point_source.py b/tests/test_point_source.py
@@ -14,7 +14,7 @@ def test_creation():
     # Ensure it has space, angle, and energy set
     assert isinstance(my_source.space, openmc.stats.Point)
     assert isinstance(my_source.angle, openmc.stats.Isotropic)
-    assert isinstance(my_source.energy, openmc.stats.muir)
+    assert isinstance(my_source.energy, openmc.stats.univariate.Normal)
 
 
 @pytest.mark.parametrize(

diff --git a/tests/test_ring_source.py b/tests/test_ring_source.py
@@ -14,7 +14,7 @@ def test_creation():
     # Ensure it has space, angle, and energy set
     assert isinstance(my_source.space, openmc.stats.CylindricalIndependent)
     assert isinstance(my_source.angle, openmc.stats.Isotropic)
-    assert isinstance(my_source.energy, openmc.stats.muir)
+    assert isinstance(my_source.energy, openmc.stats.univariate.Normal)
 
 
 @pytest.mark.parametrize("radius", [1.0, 5.6, 1e5, 7])