fix static tests

simpeg · Oct 13, 2024 · 8bdf3b1 · 8bdf3b1
1 parent 137d21a
commit 8bdf3b1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_em_static.py b/tests/test_em_static.py
@@ -1357,7 +1357,7 @@ def test_current_density(self):
 def V_from_Dipole1(
         XYZ_M, XYZ_N, rho, cur, loc_a, loc_b
 ):
-    XYZ_M = np.atleast_2d(XYZ)
+    XYZ_M = np.atleast_2d(XYZ_M)
 
     r_vec1 = XYZ_M - loc_a
     r_vec2 = XYZ_M - loc_b
@@ -1371,8 +1371,8 @@ def V_from_Dipole1(
 def V_from_Dipole2(
         XYZ_M, XYZ_N, rho, cur, loc_a, loc_b
 ):
-    XYZ_M = np.atleast_2d(XYZ)
-    XYZ_N = np.atleast_2d(XYZ_N, 3)
+    XYZ_M = np.atleast_2d(XYZ_M)
+    XYZ_N = np.atleast_2d(XYZ_N)
 
     r_vec1 = XYZ_M - loc_a
     r_vec2 = XYZ_M - loc_b
@@ -1393,7 +1393,7 @@ def V_from_Dipole2(
 def E_from_Dipole1(
         XYZ_M, XYZ_N, rho, cur, loc_a, loc_b
 ):
-    XYZ_M = np.atleast_2d(XYZ)
+    XYZ_M = np.atleast_2d(XYZ_M)
 
     r_vec1 = XYZ_M - loc_a
     r_vec2 = XYZ_M - loc_b
@@ -1407,8 +1407,8 @@ def E_from_Dipole1(
 def E_from_Dipole2(
         XYZ_M, XYZ_N, rho, cur, loc_a, loc_b
 ):
-    XYZ_M = np.atleast_2d(XYZ)
-    XYZ_N = np.atleast_2d(XYZ_N, 3)
+    XYZ_M = np.atleast_2d(XYZ_M)
+    XYZ_N = np.atleast_2d(XYZ_N)
 
     r_vec1 = XYZ_M - loc_a
     r_vec2 = XYZ_M - loc_b