test_autowrap.py 文件源码

def runtest_ufuncify(language, backend):
    has_module('numpy')
    a, b, c = symbols('a b c')
    fabc = ufuncify([a, b, c], a*b + c, language=language, backend=backend)
    facb = ufuncify([a, c, b], a*b + c, language=language, backend=backend)
    grid = numpy.linspace(-2, 2, 50)
    for b in numpy.linspace(-5, 4, 3):
        for c in numpy.linspace(-1, 1, 3):
            expected = grid*b + c
            assert numpy.sum(numpy.abs(expected - fabc(grid, b, c))) < 1e-13
            assert numpy.sum(numpy.abs(expected - facb(grid, c, b))) < 1e-13

#
# tests of language-backend combinations
#

# f2py