test_cat_query.py 文件源码

def test_query_coords(igmsp):
    # Single
    coord = SkyCoord(ra=0.0019, dec=17.7737, unit='deg')
    #
    _, _, idx = igmsp.qcat.query_coords(coord)
    assert idx[0] >= 0
    # Multiple
    coords = SkyCoord(ra=[0.0019]*2, dec=[17.7737]*2, unit='deg')
    _, _, idxs = igmsp.qcat.query_coords(coords)
    assert len(idxs) == 2
    # Dataset
    _, _, idxs2 = igmsp.qcat.query_coords(coords, groups=['BOSS_DR12'])
    assert np.sum(idxs2 >= 0) == 2
    _, _, idxs3 = igmsp.qcat.query_coords(coords, groups=['HD-LLS_DR1'])
    assert np.sum(idxs3 >= 0) == 0