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)