def test_linear(self):
# linear, ct and nearest are very inaccurate, use only for plotting!
inter = num.Interpol2D(dd=self.dd, what='linear')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt, atol=5e-1)
assert np.allclose(return_min(inter), 5.0, atol=1e-1)
assert np.allclose(inter(self.dd.XY), self.dd.zz)
dump(inter)
def test_api_bispl(self):
# API
inter = num.Interpol2D(xx=self.dd.xx,
yy=self.dd.yy,
values=self.dd.zz,
what='bispl')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0)
inter = num.Interpol2D(points=self.dd.XY,
values=self.dd.zz,
what='bispl')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0)
inter = num.Interpol2D(self.dd.XY, self.dd.zz, what='bispl')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0)
def test_ct(self):
try:
from scipy.interpolate import CloughTocher2DInterpolator
inter = num.Interpol2D(dd=self.dd, what='ct')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt, atol=1e-1)
assert np.allclose(return_min(inter), 5.0, atol=1e-1)
assert np.allclose(inter(self.dd.XY), self.dd.zz)
dump(inter)
except ImportError:
tools.skip(
"couldn't import scipy.interpolate.CloughTocher2DInterpolator")
def test_bispl(self):
# pickling this fails:
#
# ERROR: pwtools.test.test_interpol.test_interpol2d
# PicklingError: Can't pickle <type 'function'>: attribute lookup __builtin__.function failed
# FAILED (errors=1)
#
# The reason is that a function _call() inside __init__ is defined and
# pickle cannot serialize nested functions. But also defining a class-level
# function self._call_bispl instead doesn't work. So there is no solution
# here. You need to store the to-be-interpolated data and build the
# interpolator again as needed. Not a speed problem, just inconvenient. All
# other interpolator types can be pickled.
inter = num.Interpol2D(dd=self.dd, what='bispl')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0)
assert np.allclose(inter(self.dd.XY), self.dd.zz)
def test_nearest(self):
# don't even test accuracy here
inter = num.Interpol2D(dd=self.dd, what='nearest')
dump(inter)
def test_poly(self):
inter = num.Interpol2D(dd=self.dd, what='poly', deg=5)
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0, atol=1e-5)
assert np.allclose(inter(self.dd.XY), self.dd.zz)
dump(inter)
def test_rbf_inv_multi(self):
inter = num.Interpol2D(dd=self.dd, what='rbf_inv_multi')
assert np.allclose(inter([[-3, -4], [0, 0]]), self.tgt)
assert np.allclose(return_min(inter), 5.0)
assert np.allclose(inter(self.dd.XY), self.dd.zz)
dump(inter)
