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_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_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)