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))))
