test_profiles.py 文件源码

def test_mass_rescaling_yt():
    M = 6.0e14*ytu.Msun
    R = 1500.0*ytu.kpc
    pm = NFW_mass_profile()
    pd = NFW_density_profile()
    pd.set_param_values(r_s=350*ytu.kpc, rho_s=1.0*ytu.Msun/ytu.kpc**3)
    rescale_profile_by_mass(pd, "rho_s", M, R)
    pm.set_param_values(**pd.param_values)
    assert_allclose(pm(R).in_units("Msun").v, M.v)