Stop using deprecated interp2d function

GalSim-developers · Jul 17, 2023 · ff3e145 · ff3e145
1 parent 9c60812
commit ff3e145
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -407,7 +407,7 @@ def f(x_):
 def test_table2d():
     """Check LookupTable2D functionality.
     """
-    from scipy.interpolate import interp2d
+    from scipy.interpolate import RectBivariateSpline
 
     def f(x_, y_):
         return np.sin(x_) * np.cos(y_) + x_
@@ -437,8 +437,8 @@ def f(x_, y_):
                                                          for x0 in newx]
                                                         for y0 in newy]))
 
-    scitab2d = interp2d(x, y, z)
-    np.testing.assert_array_almost_equal(ref, scitab2d(newx, newy))
+    scitab2d = RectBivariateSpline(x, y, z.T, kx=1, ky=1)
+    np.testing.assert_array_almost_equal(ref, scitab2d(newx, newy).T)
 
     # Try using linear GSInterp
     tab2d2 = galsim.LookupTable2D(x, y, z, interpolant=galsim.Linear())
@@ -466,8 +466,8 @@ def f(x_, y_):
     np.testing.assert_array_almost_equal(ref, np.array([[tab2d(x0, y0)
                                                          for x0 in newx]
                                                         for y0 in newy]))
-    scitab2d = interp2d(x, y, z)
-    np.testing.assert_array_almost_equal(ref, scitab2d(newx, newy))
+    scitab2d = RectBivariateSpline(x, y, z.T, kx=1, ky=1)
+    np.testing.assert_array_almost_equal(ref, scitab2d(newx, newy).T)
 
     # Using a galsim.Interpolant should raise an exception if x/y are not equal spaced.
     with assert_raises(galsim.GalSimIncompatibleValuesError):