def test_set_ignore_mean(): rv = RV() assert rv.ignore_mean is False rv.ignore_mean = True assert rv.ignore_mean is True rv.ignore_mean = False assert rv.ignore_mean is False
def test_RV_to_dict_updates_parameters_in_params(): rv = RV(semi_amp=1.0, period=3, tau=4, omega=5, ecc=0.5, mean_val=8, ignore_mean=True) assert rv.ignore_mean == True assert rv._params["ignore_mean"] == True assert rv.semi_amp == 1.0 # _params not updated (yet) rv.semi_amp = 2 rv.ignore_mean = False assert rv._params["k1"] == 1 assert rv._params["ignore_mean"] == True assert rv.ignore_mean == False # RV.to_dict() updates _params param_dict = rv.to_dict() assert param_dict["ignore_mean"] == False assert rv._params["ignore_mean"] == False assert rv.semi_amp == 2 assert param_dict["k1"] == 2 assert rv._params["k1"] == 2