diff --git a/tests/test_point_source.py b/tests/test_point_source.py index 53fc42e..74f0b8d 100644 --- a/tests/test_point_source.py +++ b/tests/test_point_source.py @@ -15,9 +15,11 @@ def test_creation(): # Ensure it has space, angle, and energy set assert isinstance(source.space, openmc.stats.Point) 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) + 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 25e757a..57c8ba8 100644 --- a/tests/test_ring_source.py +++ b/tests/test_ring_source.py @@ -15,9 +15,11 @@ def test_creation(): # Ensure it has space, angle, and energy set assert isinstance(source.space, openmc.stats.CylindricalIndependent) 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) + 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])