def test_wcsgroupcat_recalc_catalog_radec(mock_fits_wcs, rect_imcat):
ra, dec = mock_fits_wcs.all_pix2world(rect_imcat.catalog['x'],
rect_imcat.catalog['y'], 0)
refcat = Table([ra, dec], names=('RA', 'DEC'))
ref = RefCatalog(refcat)
wim1 = copy.deepcopy(rect_imcat)
wim2 = copy.deepcopy(rect_imcat)
g = WCSGroupCatalog(wim1)
ra = g.catalog['RA']
dec = g.catalog['DEC']
# artificially add another image:
g._images.append(wim2)
g.recalc_catalog_radec()
assert np.allclose(g.catalog['RA'], ra)
assert np.allclose(g.catalog['DEC'], dec)
g.align_to_ref(ref)
mcat = g.get_matched_cat()
assert len(mcat) == 5
assert np.allclose(mcat['RA'], ra)
assert np.allclose(mcat['DEC'], dec)
del g.catalog['RA']
del g.catalog['DEC']
with pytest.raises(RuntimeError):
g.calc_tanp_xy(None)
def test_wcsgroupcat_match2ref(mock_fits_wcs, rect_imcat):
ra, dec = mock_fits_wcs.all_pix2world(rect_imcat.catalog['x'],
rect_imcat.catalog['y'], 0)
refcat = Table([ra[:-1], dec[:-1]], names=('RA', 'DEC'))
ref = RefCatalog(refcat)
# unequal catalog lengths
g = WCSGroupCatalog(rect_imcat)
with pytest.raises(ValueError):
g.match2ref(ref, match=None)
refcat = Table([ra, dec], names=('RA', 'DEC'))
ref = RefCatalog(refcat)
# call calc_tanp_xy before matching
with pytest.raises(RuntimeError) as e:
g.match2ref(ref, match=TPMatch())
assert (e.value.args[0] == "'calc_tanp_xy()' should have been run "
"prior to match2ref()")
ref.calc_tanp_xy(rect_imcat.tpwcs)
g.calc_tanp_xy(rect_imcat.tpwcs)
g.catalog['matched_ref_id'] = np.ones(5, dtype=bool)
g.catalog['_raw_matched_ref_idx'] = np.ones(5, dtype=bool)
g.match2ref(ref, match=TPMatch())
def test_wcsrefcat_intersections(mock_fits_wcs, rect_imcat):
ra, dec = mock_fits_wcs.all_pix2world(10 * rect_imcat.catalog['x'],
10 * rect_imcat.catalog['y'], 0)
refcat = Table([ra, dec], names=('RA', 'DEC'))
ref = RefCatalog(refcat)
ref.calc_tanp_xy(rect_imcat.tpwcs)
pts1 = list(ref.polygon.points)[0]
pts2 = list(ref.intersection(ref.polygon).points)[0]
for pt1 in pts1:
assert any(np.allclose(pt1, pt2) for pt2 in pts2)
pts2 = list(ref.intersection(ref).points)[0]
for pt1 in pts1:
assert any(np.allclose(pt1, pt2) for pt2 in pts2)
assert np.allclose(
ref.intersection_area(ref), 2.9902125220360176e-10, atol=0.0,
rtol=0.005,
)