from pint.models.binary_dd import BinaryDD from pint.models.binary_ddk import BinaryDDK from pint.models.binary_ell1 import BinaryELL1, BinaryELL1H from pint.models.dispersion_model import DispersionDM, DispersionDMX from pint.models.frequency_dependent import FD from pint.models.glitch import Glitch from pint.models.jump import DelayJump, PhaseJump from pint.models.model_builder import get_model from pint.models.noise_model import EcorrNoise, PLRedNoise, ScaleToaError from pint.models.solar_system_shapiro import SolarSystemShapiro from pint.models.solar_wind_dispersion import SolarWindDispersion from pint.models.spindown import Spindown # Import the main timing model classes from pint.models.timing_model import TimingModel, DEFAULT_ORDER from pint.models.wave import Wave from pint.models.ifunc import IFunc # Define a standard basic model StandardTimingModel = TimingModel( "StandardTimingModel", [AstrometryEquatorial(), Spindown(), DispersionDM(), SolarSystemShapiro()], ) # BTTimingModel = generate_timing_model("BTTimingModel", # (Astrometry, Spindown, Dispersion, SolarSystemShapiro, BT)) # DDTimingModel = generate_timing_model("DDTimingModel", # (Astrometry, Spindown, Dispersion, SolarSystemShapiro, DD))
def test_forgot_name(): """Check argument validation in case 'name' is forgotten.""" with pytest.raises(ValueError): TimingModel(AstrometryEquatorial()) with pytest.raises(ValueError): TimingModel([AstrometryEquatorial(), DispersionDM()])
def model(): """Make a simple model.""" return TimingModel( components=[AstrometryEquatorial(), DispersionDM(), DispersionDMX()] )
# Import all standard model components here from pint.models.astrometry import AstrometryEcliptic, AstrometryEquatorial from pint.models.binary_bt import BinaryBT from pint.models.binary_dd import BinaryDD from pint.models.binary_ddk import BinaryDDK from pint.models.binary_ell1 import BinaryELL1, BinaryELL1H from pint.models.dispersion_model import DispersionDM, DispersionDMX from pint.models.frequency_dependent import FD from pint.models.glitch import Glitch from pint.models.jump import DelayJump, PhaseJump from pint.models.model_builder import get_model from pint.models.noise_model import EcorrNoise, PLRedNoise, ScaleToaError from pint.models.solar_system_shapiro import SolarSystemShapiro from pint.models.solar_wind_dispersion import SolarWindDispersion from pint.models.spindown import Spindown # Import the main timing model classes from pint.models.timing_model import TimingModel from pint.models.wave import Wave from pint.models.ifunc import IFunc # Define a standard basic model StandardTimingModel = TimingModel( "StandardTimingModel", (AstrometryEquatorial(), Spindown(), DispersionDM(), SolarSystemShapiro()), ) # BTTimingModel = generate_timing_model("BTTimingModel", # (Astrometry, Spindown, Dispersion, SolarSystemShapiro, BT)) # DDTimingModel = generate_timing_model("DDTimingModel", # (Astrometry, Spindown, Dispersion, SolarSystemShapiro, DD))