def test_shift_coords(self):
numPix = 10
deltaPix = 0.05
x_grid, y_grid, ra_at_xy_0, dec_at_xy_0, x_at_radec_0, y_at_radec_0, Mpix2coord, Mcoord2pix = util.make_grid_with_coordtransform(
numPix=numPix, deltapix=deltaPix, subgrid_res=1)
# mask (1= model this pixel, 0= leave blanck)
kwargs_data = {
'ra_at_xy_0': ra_at_xy_0,
'dec_at_xy_0': dec_at_xy_0,
'transform_pix2angle': Mpix2coord,
'image_data': np.ones((numPix, numPix))
}
data = Data(kwargs_data)
ra_shift = 0.05
dec_shift = 0.
kwargs_data['ra_shift'] = ra_shift
kwargs_data['dec_shift'] = dec_shift
data_shift = Data(kwargs_data)
ra, dec = data.map_pix2coord(1, 1)
ra_new, dec_new = data_shift.map_pix2coord(1, 1)
npt.assert_almost_equal(ra_new - ra, ra_shift, decimal=10)
npt.assert_almost_equal(dec_new - dec, dec_shift, decimal=10)
ra_2, dec_2 = data_shift.map_pix2coord(0, 1)
npt.assert_almost_equal(ra, ra_2, decimal=10)
npt.assert_almost_equal(dec, dec_2, decimal=10)
x, y = data.map_coord2pix(0, 0)
x_new, y_new = data_shift.map_coord2pix(ra_shift, dec_shift)
npt.assert_almost_equal(x, x_new, decimal=10)
npt.assert_almost_equal(y, y_new, decimal=10)
def test_shift_coordinate_grid(self):
x_shift = 0.05
y_shift = 0
kwargs_data_shifted = self.SimAPI.shift_coordinate_grid(
self.kwargs_data, x_shift, y_shift, pixel_units=False)
kwargs_data_new = copy.deepcopy(self.kwargs_data)
kwargs_data_new['ra_shift'] = x_shift
kwargs_data_new['dec_shift'] = y_shift
data = Data(kwargs_data=kwargs_data_shifted)
data_new = Data(kwargs_data=kwargs_data_new)
ra, dec = 0, 0
x, y = data.map_coord2pix(ra, dec)
x_new, y_new = data_new.map_coord2pix(ra, dec)
npt.assert_almost_equal(x, x_new, decimal=10)
npt.assert_almost_equal(y, y_new, decimal=10)
ra, dec = data.map_pix2coord(x, y)
ra_new, dec_new = data_new.map_pix2coord(x, y)
npt.assert_almost_equal(ra, ra_new, decimal=10)
npt.assert_almost_equal(dec, dec_new, decimal=10)
x_coords, y_coords = data.coordinates
x_coords_new, y_coords_new = data_new.coordinates
npt.assert_almost_equal(x_coords[0], x_coords_new[0], decimal=10)
npt.assert_almost_equal(y_coords[0], y_coords_new[0], decimal=10)