test_enr2.py 文件源码

def test_integral0(n=4):
    '''Make sure that the polynomials are orthonormal
    '''
    x = sympy.Symbol('x')
    y = sympy.Symbol('y')
    z = sympy.Symbol('z')
    vals = numpy.concatenate(
        orthopy.enr2.tree(n, numpy.array([x, y, z]), symbolic=True)
        )

    assert sympy.integrate(
        vals[0] * sympy.exp(-x**2-y**2-z**2),
        (x, -oo, +oo), (y, -oo, +oo), (z, -oo, +oo)
        ) == sympy.sqrt(sympy.sqrt(sympy.pi))**3
    for val in vals[1:]:
        assert sympy.integrate(
            val * sympy.exp(-x**2-y**2-z**2),
            (x, -oo, +oo), (y, -oo, +oo), (z, -oo, +oo)
            ) == 0
    return