Test mag grad components on singular points

Test if evaluation of the mag grad components on singular points return nans.
fatiando · Oct 3, 2024 · d1fa434 · d1fa434
1 parent a6b70ac
commit d1fa434
Showing 1 changed file with 16 additions and 9 deletions.
diff --git 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