def test_output(self): x = rd.random(self.n) y = rd.random(self.n) interp_values = math.lin_interp(x, y, self.k) self.assertEqual(np.shape(interp_values), (self.n, self.k + 2)) interp_diff = np.diff(interp_values) for i in xrange(self.k + 1): self.assertTrue(np.allclose(interp_diff[:, 0], interp_diff[:, i]))
def test_same_values(self): x = rd.random(self.n) interp_values = math.lin_interp(x, x, self.k) diff = interp_values - np.array([x] * (self.k + 2)).T self.assertTrue(np.allclose(diff, np.zeros((self.n, self.k + 2))))