def test_ftest_wb():
"""Test for wideband fitter class F-test."""
wb_m = tm.get_model(os.path.join(datadir, "J0023+0923_ell1_simple.par"))
wb_t = toa.make_fake_toas(
56000.0, 56001.0, 10, wb_m, freq=1400.0, obs="GBT", dm=wb_m.DM.value
)
wb_f = fitter.WidebandTOAFitter(wb_t, wb_m)
wb_f.fit_toas()
# Parallax
PX = param.floatParameter(
parameter_type="float", name="PX", value=0.0, units=u.mas, frozen=False
)
PX_Component = "AstrometryEcliptic"
# A1DOT
A1DOT = param.floatParameter(
parameter_type="float",
name="A1DOT",
value=0.0,
units=ls / u.second,
frozen=False,
)
A1DOT_Component = "BinaryELL1"
# Test adding A1DOT
Ftest_dict = wb_f.ftest(A1DOT, A1DOT_Component, remove=False, full_output=True)
assert isinstance(Ftest_dict["ft"], float) or isinstance(Ftest_dict["ft"], bool)
# Test removing parallax
Ftest_dict = wb_f.ftest(PX, PX_Component, remove=True, full_output=True)
assert isinstance(Ftest_dict["ft"], float) or isinstance(Ftest_dict["ft"], bool)
def test_ftest_wb():
"""Test for wideband fitter class F-test."""
wb_m = tm.get_model(
os.path.join(datadir, "J1614-2230_NANOGrav_12yv3.wb.gls.par"))
wb_t = toa.get_TOAs(
os.path.join(datadir, "J1614-2230_NANOGrav_12yv3.wb.tim"))
wb_f = fitter.WidebandTOAFitter(wb_t, wb_m)
wb_f.fit_toas()
# Parallax
PX = param.floatParameter(parameter_type="float",
name="PX",
value=0.0,
units=u.mas,
frozen=False)
PX_Component = "AstrometryEcliptic"
# A1DOT
A1DOT = param.floatParameter(
parameter_type="float",
name="A1DOT",
value=0.0,
units=ls / u.second,
frozen=False,
)
A1DOT_Component = "BinaryELL1"
# Test adding A1DOT
Ftest_dict = wb_f.ftest(A1DOT,
A1DOT_Component,
remove=False,
full_output=True)
assert isinstance(Ftest_dict["ft"], float) or isinstance(
Ftest_dict["ft"], bool)
# Test removing parallax
Ftest_dict = wb_f.ftest(PX, PX_Component, remove=True, full_output=True)
assert isinstance(Ftest_dict["ft"], float) or isinstance(
Ftest_dict["ft"], bool)