def test__interp_between_vectors(N): index = numpy.arange(0, 4) vector1 = -1 * index**2 - 1 vector2 = 2 * index**2 + 2 expected = { 1: numpy.array([ [ -1.0, -2.0, -5.0, -10.0], [ 0.5, 1.0, 2.5, 5.0], [ 2.0, 4.0, 10.0, 20.0], ]), 3: numpy.array([ [ -1.00, -2.00, -5.00, -10.00], [ -0.25, -0.50, -1.25, -2.50], [ 0.50, 1.00, 2.50, 5.00], [ 1.25, 2.50, 6.25, 12.50], [ 2.00, 4.00, 10.00, 20.00], ]) } if N: result = core._interp_between_vectors(vector1, vector2, n_nodes=N) nptest.assert_array_equal(result, expected[N]) else: with raises(ValueError): core._interp_between_vectors(vector1, vector2, n_nodes=0)
def test_bad_points(self): core._interp_between_vectors(self.vector1, self.vector2, n_points=0)
def test_insert_3(self): result = core._interp_between_vectors(self.vector1, self.vector2, n_points=3) nptest.assert_array_equal(result, self.known_insert_3)