test_lambdify.py 文件源码

def test_issue9474():
    mods = [None, 'math']
    if numpy:
        mods.append('numpy')
    if mpmath:
        mods.append('mpmath')
    for mod in mods:
        f = lambdify(x, sympy.S(1)/x, modules=mod)
        assert f(2) == 0.5
        f = lambdify(x, floor(sympy.S(1)/x), modules=mod)
        assert f(2) == 0

    if mpmath:
        f = lambdify(x, sympy.S(1)/sympy.Abs(x), modules=['mpmath'])
        assert isinstance(f(2), mpmath.mpf)

    for absfunc, modules in product([Abs, abs], mods):
        f = lambdify(x, absfunc(x), modules=modules)
        assert f(-1) == 1
        assert f(1) == 1
        assert f(3+4j) == 5