def test__vector_apply(): def f(x, y): return x * x + y y = 10 a = np.array([1., 2., 3.]) expected = [f(x, y) for x in a] va = _vector_apply(f, a, tuple([y])) assert_allclose(va, expected)
def test__vector_apply(): def f(x, y): return x * x + y y = 10 a = np.array([1., 2., 3.]) expected = [f(x, y) for x in a] va = _vector_apply(f, a, tuple([y])) for i in range(len(a)): assert_almost_equal(va[i], expected[i])