test_sphere_sph.py 文件源码

def test_integral0(n=4):
    polar = sympy.Symbol('theta', real=True)
    azimuthal = sympy.Symbol('phi', real=True)
    tree = numpy.concatenate(
            orthopy.sphere.sph_tree(
                n, polar, azimuthal, normalization='quantum mechanic',
                symbolic=True
                ))

    assert sympy.integrate(
        tree[0] * sympy.sin(polar), (polar, 0, pi), (azimuthal, 0, 2*pi)
        ) == 2*sqrt(pi)
    for val in tree[1:]:
        assert sympy.integrate(
            val * sympy.sin(polar), (azimuthal, 0, 2*pi), (polar, 0, pi)
            ) == 0
    return