test_math.py 文件源码

def test_pow():
    n_x = f_x**3
    assert n_x.ndim == 1
    assert isinstance(n_x, Formula1D)
    assert_allclose(n_x(x).v, (f_x(x)**3).v)
    assert str(n_x(x).units) == "km**3"
    m_x = f_x**0.5
    assert n_x.ndim == 1
    assert isinstance(m_x, Formula1D)
    assert_allclose(m_x(x).v, (f_x(x)**0.5).v)
    assert str(m_x(x).units) == "sqrt(km)"
    l_x = f_x**Float(0.3)
    assert l_x.ndim == 1
    assert isinstance(l_x, Formula1D)
    assert_allclose(l_x(x).v, (f_x(x)**0.3).v)
    assert str(l_x(x).units) == "km**(3/10)"