def test_add_DMX():
""" Check that the DMX model contains the DMX events after they are added. """
dm_mod = DispersionDMX()
index = 3
dmx = 1.0
mjd_start = 58000.0
mjd_end = 58100.0
init_len = len(dm_mod.params)
dm_mod.add_DMX_range(mjd_start, mjd_end, index, dmx, frozen=False)
assert len(dm_mod.params) == init_len + 3
nm = "DMX_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == dmx
nm = "DMXR1_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == mjd_start
nm = "DMXR2_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == mjd_end
def test_only_one_MJD():
""" Check for error when one of the MJDs is None and the other isn't. """
dm_mod = DispersionDMX()
with pytest.raises(ValueError):
dm_mod.add_DMX_range(None, 58000, 3, 1, frozen=False)
with pytest.raises(ValueError):
dm_mod.add_DMX_range(58000, None, 3, 1, frozen=False)
def test_print_inds():
dm_mod = DispersionDMX()
dmx = 1.0
mjd_start = 58000.0
mjd_end = 58100.0
inds_init = 5
dm_mod.remove_DMX_range(1)
for i in range(inds_init):
dm_mod.add_DMX_range(mjd_start, mjd_end, i, dmx, frozen=False)
inds = dm_mod.get_indices()
assert len(inds) == inds_init
for i in range(inds_init):
assert i in inds
def test_remove_DMX():
"""Check the DMX model no longer has the DMX components after they are removed. """
dm_mod = DispersionDMX()
index = 3
dmx = 1.0
mjd_start = 58000.0
mjd_end = 58100.0
dm_mod.add_DMX_range(mjd_start, mjd_end, index, dmx, frozen=False)
dm_mod.remove_DMX_range(index)
for pn in ["DMX_", "DMXR1_", "DMXR2_"]:
nm = str(pn) + str(f"{int(index):04d}")
assert nm not in dm_mod.params
def test_dynamic_index():
dm_mod = DispersionDMX()
index1 = 3
dmx1 = 1.0
mjd_start1 = 58000.0
mjd_end1 = 58100.0
index2 = None
dmx2 = 1.0
mjd_start2 = 58200.0
mjd_end2 = 58300.0
dm_mod.add_DMX_range(mjd_start1, mjd_end1, index1, dmx1, frozen=False)
dct = dm_mod.get_prefix_mapping_component("DMX_")
ind = np.max(list(dct.keys())) + 1
newind = dm_mod.add_DMX_range(mjd_start2,
mjd_end2,
index2,
dmx2,
frozen=False)
assert newind == ind
def test_unusual_index():
""" Check for appropriate results for negative and floating point indices. """
dm_mod = DispersionDMX()
dmx = 1.0
mjd_start = 58000.0
mjd_end = 58100.0
index = -1
with pytest.raises(PrefixError):
dm_mod.add_DMX_range(mjd_start, mjd_end, index, dmx, frozen=False)
index = 2.0
init_len = len(dm_mod.params)
dm_mod.add_DMX_range(mjd_start, mjd_end, index, dmx, frozen=False)
assert len(dm_mod.params) == init_len + 3
nm = "DMX_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == dmx
nm = "DMXR1_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == mjd_start
nm = "DMXR2_" + f"{int(index):04d}"
comp = getattr(dm_mod, nm)
assert comp.value == mjd_end
def test_nonetype_MJD():
dm_mod = DispersionDMX()
with pytest.raises(ValueError):
dm_mod.add_DMX_range(None, 58000, 3, 1, frozen=False)
with pytest.raises(ValueError):
dm_mod.add_DMX_range(58000, None, 3, 1, frozen=False)
def test_remove_nonexistent_index():
""" Check for error when a unused DMX index is removed. """
dm_mod = DispersionDMX()
index = 3
with pytest.raises(ValueError):
dm_mod.remove_DMX_range(index)
def test_duplicate_index():
""" Check for error when a duplicate DMX index is used. """
dm_mod = DispersionDMX()
dm_mod.add_DMX_range(58000, 58100, 3, 1, frozen=False)
with pytest.raises(ValueError):
dm_mod.add_DMX_range(58200, 58300, 3, 1, frozen=False)
def test_startMJD_greaterThan_endMJD():
""" Check for error when start MJD is larger than end MJD. """
dm_mod = DispersionDMX()
with pytest.raises(ValueError):
dm_mod.add_DMX_range(58100, 58000, 3, 1, frozen=False)
def model():
"""Make a simple model."""
return TimingModel(
components=[AstrometryEquatorial(), DispersionDM(), DispersionDMX()]
)