diff --git a/tests/test_point_source.py b/tests/test_point_source.py index 368272d..53fc42e 100644 --- a/tests/test_point_source.py +++ b/tests/test_point_source.py @@ -17,7 +17,7 @@ def test_creation(): assert isinstance(source.angle, openmc.stats.Isotropic) assert isinstance(source.energy, openmc.stats.univariate.Normal) or isinstance( source.energy, openmc.stats.univariate.Tabular - ) + ) or isinstance(source.energy, openmc.stats.Mixture) @pytest.mark.parametrize( diff --git a/tests/test_ring_source.py b/tests/test_ring_source.py index e7a0008..25e757a 100644 --- a/tests/test_ring_source.py +++ b/tests/test_ring_source.py @@ -17,7 +17,7 @@ def test_creation(): assert isinstance(source.angle, openmc.stats.Isotropic) assert isinstance(source.energy, openmc.stats.univariate.Normal) or isinstance( source.energy, openmc.stats.univariate.Tabular - ) + ) or isinstance(source.energy, openmc.stats.Mixture) @pytest.mark.parametrize("radius", [1, 5.6, 1e5, 7.0])