def test_roundtrip_vct():
varying_matrix_lt = [
rotation_matrix(a)[:2, :2] for a in np.linspace(0, 90, 10)
] * u.arcsec
vct = VaryingCelestialTransform(crpix=(5, 5) * u.pix,
cdelt=(1, 1) * u.arcsec / u.pix,
crval_table=(0, 0) * u.arcsec,
pc_table=varying_matrix_lt,
lon_pole=180 * u.deg)
new_vct = roundtrip_object(vct)
assert isinstance(new_vct, VaryingCelestialTransform)
new_ivct = roundtrip_object(vct.inverse)
assert isinstance(new_ivct, InverseVaryingCelestialTransform)
assert u.allclose(u.Quantity(new_vct.crpix), (5, 5) * u.pix)
assert u.allclose(u.Quantity(new_ivct.crpix), (5, 5) * u.pix)
assert u.allclose(u.Quantity(new_vct.pc_table), varying_matrix_lt)
assert u.allclose(u.Quantity(new_ivct.pc_table), varying_matrix_lt)
pixel = (0 * u.pix, 0 * u.pix, 5 * u.pix)
world = new_vct(*pixel)
assert u.allclose(world, (359.99804329 * u.deg, 0.00017119 * u.deg))
assert u.allclose(new_ivct(*world, 5 * u.pix),
pixel[:2],
atol=0.01 * u.pix)
def test_roundtrip_vct_2d():
varying_matrix_lt = [rotation_matrix(a)[:2, :2] for a in np.linspace(0, 90, 15)] * u.arcsec
varying_matrix_lt = varying_matrix_lt.reshape((5, 3, 2, 2))
vct = VaryingCelestialTransform2D(crpix=(5, 5) * u.pix,
cdelt=(1, 1) * u.arcsec/u.pix,
crval_table=(0, 0) * u.arcsec,
pc_table=varying_matrix_lt,
lon_pole=180 * u.deg)
new_vct = roundtrip_object(vct)
assert isinstance(new_vct, VaryingCelestialTransform2D)
new_ivct = roundtrip_object(vct.inverse)
assert isinstance(new_ivct, InverseVaryingCelestialTransform2D)
assert u.allclose(u.Quantity(new_vct.crpix), (5, 5) * u.pix)
assert u.allclose(u.Quantity(new_ivct.crpix), (5, 5) * u.pix)
assert u.allclose(u.Quantity(new_vct.pc_table), varying_matrix_lt)
assert u.allclose(u.Quantity(new_ivct.pc_table), varying_matrix_lt)
pixel = (0*u.pix, 0*u.pix, 4*u.pix, 2*u.pix)
world = new_vct(*pixel)
assert u.allclose(world, (359.99861111, 0.00138889)*u.deg)
assert u.allclose(new_ivct(*world, 4*u.pix, 2*u.pix), pixel[:2], atol=0.01*u.pix)
def test_roundtrip_vct_slit2d():
varying_matrix_lt = [rotation_matrix(a)[:2, :2] for a in np.linspace(0, 90, 15)] * u.arcsec
varying_matrix_lt = varying_matrix_lt.reshape((5, 3, 2, 2))
vct = VaryingCelestialTransformSlit2D(crpix=(5, 5) * u.pix,
cdelt=(1, 1) * u.arcsec/u.pix,
crval_table=(0, 0) * u.arcsec,
pc_table=varying_matrix_lt,
lon_pole=180 * u.deg)
new_vct = roundtrip_object(vct)
assert isinstance(new_vct, VaryingCelestialTransformSlit2D)
new_ivct = roundtrip_object(vct.inverse)
assert isinstance(new_ivct, InverseVaryingCelestialTransformSlit2D)
def test_roundtrip_tiled_dataset(simple_tiled_dataset):
newobj = roundtrip_object(simple_tiled_dataset)
assert simple_tiled_dataset.inventory == newobj.inventory
for old_ds, new_ds in zip(simple_tiled_dataset.flat, newobj.flat):
assert_dataset_equal(new_ds, old_ds)
def test_coupled_compound_model():
ccm = CoupledCompoundModel("&", m.Shift(5), m.Scale(10))
new = roundtrip_object(ccm)
assert isinstance(new, CoupledCompoundModel)
assert isinstance(new.left, m.Shift)
assert isinstance(new.right, m.Scale)
assert ccm.n_inputs == new.n_inputs
assert ccm.inputs == new.inputs
def test_roundtrip_dataset(dataset):
newobj = roundtrip_object(dataset)
assert_dataset_equal(newobj, dataset)
def test_roundtrip_file_manager(file_manager):
newobj = roundtrip_object(file_manager)
assert newobj == file_manager