test_cat_query.py 文件源码

def test_cat_from_query_coord(igmsp):
    # Single
    coord = SkyCoord(ra=0.0019, dec=17.7737, unit='deg')
    #
    _, ccat, _ = igmsp.qcat.query_coords(coord)
    assert ccat['IGM_ID'][0] == 0
    # Multiple
    coords = SkyCoord(ra=[0.0028,0.0019], dec=[14.9747,17.7737], unit='deg')
    _, ccat2, _ = igmsp.qcat.query_coords(coords)
    assert len(ccat2) == 2
    assert ccat2['IGM_ID'][0] == 1
    assert ccat2['IGM_ID'][1] == 0
    # One miss
    coords3 = SkyCoord(ra=[9.0028,0.0019], dec=[-14.9747,17.7737], unit='deg')
    _, ccat3, _ = igmsp.qcat.query_coords(coords3)
    assert ccat3['IGM_ID'][0] == -1