test_profiles.py 文件源码

def test_mass_rescaling_astropy():
    M = 6.0e14*apu.Msun
    R = 1500.0*apu.kpc
    pm = NFW_mass_profile()
    pd = NFW_density_profile()
    pd.set_param_values(r_s=350*apu.kpc, rho_s=1.0*apu.Msun/apu.kpc**3)
    rescale_profile_by_mass(pd, "rho_s", M, R)
    pm.set_param_values(**pd.param_values)
    assert_allclose(pm(R).to("Msun").value, M.value)