diff --git a/choclo/tests/test_prism_magnetic.py b/choclo/tests/test_prism_magnetic.py index 22f77828..f71c248c 100644 --- a/choclo/tests/test_prism_magnetic.py +++ b/choclo/tests/test_prism_magnetic.py @@ -676,6 +676,19 @@ class TestMagneticFieldSingularities: we approach from outside of the prism. """ + COMPONENTS = ( + magnetic_field, + magnetic_e, + magnetic_n, + magnetic_u, + magnetic_ee, + magnetic_en, + magnetic_eu, + magnetic_nn, + magnetic_nu, + magnetic_uu, + ) + @pytest.fixture() def sample_prism(self): """ @@ -752,9 +765,7 @@ def get_interior_points(self, prism): coordinates = tuple(c.ravel() for c in np.meshgrid(easting, northing, upward)) return coordinates - @pytest.mark.parametrize( - "forward_func", (magnetic_field, magnetic_e, magnetic_n, magnetic_u) - ) + @pytest.mark.parametrize("forward_func", COMPONENTS) def test_on_vertices(self, sample_prism, forward_func): """ Test if magnetic field components on vertices are equal to NaN @@ -767,9 +778,7 @@ def test_on_vertices(self, sample_prism, forward_func): ) assert np.isnan(results).all() - @pytest.mark.parametrize( - "forward_func", (magnetic_field, magnetic_e, magnetic_n, magnetic_u) - ) + @pytest.mark.parametrize("forward_func", COMPONENTS) @pytest.mark.parametrize("direction", ("easting", "northing", "upward")) def test_on_edges(self, sample_prism, direction, forward_func): """ @@ -785,9 +794,7 @@ def test_on_edges(self, sample_prism, direction, forward_func): ) assert np.isnan(results).all() - @pytest.mark.parametrize( - "forward_func", (magnetic_field, magnetic_e, magnetic_n, magnetic_u) - ) + @pytest.mark.parametrize("forward_func", COMPONENTS) def test_on_interior_points(self, sample_prism, forward_func): """ Test if magnetic field components are NaN on internal points