Verbose variable name for direction of finite diff

fatiando · Oct 4, 2024 · aaf6f65 · aaf6f65
1 parent d10cf70
commit aaf6f65
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/choclo/tests/test_prism_magnetic.py b/choclo/tests/test_prism_magnetic.py
@@ -889,6 +889,7 @@ def test_derivatives_of_magnetic_components(
         # Get forward functions
         mag_func = get_forward_function(i)
         mag_grad_func = get_forward_function(i + j)
+        direction = j  # direction along which to apply the finite differences
         # Evaluate the mag grad function on the sample grid
         mag_grad = evaluate(
             mag_grad_func, sample_3d_grid, sample_prism, sample_magnetization
@@ -898,7 +899,7 @@ def test_derivatives_of_magnetic_components(
             sample_3d_grid,
             sample_prism,
             sample_magnetization,
-            direction=j,
+            direction=direction,
             forward_func=mag_func,
             delta=self.delta,
         )
@@ -963,6 +964,7 @@ def test_derivatives_of_magnetic_components_on_edges(
         # Get forward functions
         mag_func = get_forward_function(i)
         mag_grad_func = get_forward_function(f"{i}{j}")
+        direction = j  # direction along which to apply the finite differences
         # Evaluate the mag grad function on the sample grid
         mag_grad = evaluate(
             mag_grad_func, coordinates, sample_prism, sample_magnetization
@@ -972,7 +974,7 @@ def test_derivatives_of_magnetic_components_on_edges(
             coordinates,
             sample_prism,
             sample_magnetization,
-            direction=j,
+            direction=direction,
             forward_func=mag_func,
             delta=self.delta,
         )