Python _vector_apply Examples

Example #1

File: test_func.py Project: vincecr0ft/probfit

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)

Example #2

File: testfunc.py Project: bks/probfit

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

Example #3

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