def test_set_observer(self): """Test set_observer.""" observer1 = MockObserver("Obs1") observer2 = MockObserver("Obs2") pts = ((0, 0, 0), (10, 0, 1), (10, 10, 2), (0, 10, 3)) tris = (0, 1, 3, 1, 2, 3) interp = InterpIdw(pts, tris) # Non-Observer Type with self.assertRaises(TypeError) as context: interp.set_observer("xyz") err = context.exception self.assertIn("SetObserver(): incompatible function arguments.", str(err)) interp.set_observer(observer1) interp.interpolate_to_points(np.random.rand(100000, 3) + 5) self.assertGreater(observer1.status['percent_complete'], 0.0) self.assertGreater(observer1.status['elapsed_seconds'], 0.0) interp.set_observer(observer2) interp.interpolate_to_points(np.random.rand(100000, 3) + 5) self.assertGreater(observer2.status['percent_complete'], 0.0) self.assertGreater(observer2.status['elapsed_seconds'], 0.0)
def test_interp_to_pts(self): """Interpolate to multiple points.""" pts = ((0, 0, 0), (10, 0, 1), (10, 10, 2), (0, 10, 3)) interp = InterpIdw(pts) ret = interp.interpolate_to_points( np.array([(2.0, 1.0, 0.0), (5.0, 10.0, 2.5)])) np.testing.assert_array_almost_equal((0.02681550197303295, 2.5), ret)
def test_interp_to_pts_numpy(self): """Interpolate to multiple points.""" pts = np.array([(0, 0, 0), (10, 0, 1), (10, 10, 2), (0, 10, 3)]) interp = InterpIdw(pts) ret = interp.interpolate_to_points( np.array(((2.0, 1.0, 0.0), (5.0, 10.0, 2.5)))) self.assertIsInstance(ret, np.ndarray) np.testing.assert_array_equal(np.array([0.02681550197303295, 2.5]), ret)