def test_interpolate_polar(self): data = np.arange(12.0).reshape(4, 3) masked_values = (data == 2) | (data == 9) filled_a = ipol.interpolate_polar(data, mask=masked_values, ipclass=ipol.Linear) testfunc = ipol.interpolate_polar with pytest.raises(ipol.MissingTargetsError): testfunc(data, mask=None, ipclass=ipol.Linear) mdata = np.ma.array(data, mask=masked_values) filled_b = ipol.interpolate_polar(mdata, ipclass=ipol.Linear) np.testing.assert_allclose(filled_a, filled_b)
def test_interpolate_polar(self): data = np.arange(12.).reshape(4, 3) masked_values = (data == 2) | (data == 9) filled_a = ipol.interpolate_polar(data, mask=masked_values, ipclass=ipol.Linear) testfunc = ipol.interpolate_polar self.assertRaises(ipol.MissingTargetsError, lambda: testfunc(data, mask=None, ipclass=ipol.Linear)) mdata = np.ma.array(data, mask=masked_values) filled_b = ipol.interpolate_polar(mdata, ipclass=ipol.Linear) np.testing.assert_allclose(filled_a, filled_b)