示例#1
0
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)
示例#2
0
 def test_bad_points(self):
     core._interp_between_vectors(self.vector1, self.vector2, n_points=0)
示例#3
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)