def test_list_pagination(self, simple_populated_db, reset_globals_to_defaults): """ Sanity check that pagination params passed correctly to wrapper func. (NB wrapper tested independently elsewhere). """ vr.default_list_n_max = 30 vr.default_pagesize = 5 dbinf = simple_populated_db count = apiv1.count() assert count > vr.default_list_n_max ivorns = apiv1.list_ivorn() assert len(ivorns) == vr.default_list_n_max # This time we override the max: ivorns = apiv1.list_ivorn(n_max=0) assert len(ivorns) == dbinf.n_inserts
import voeventdb.remote.apiv1 as apiv1 from astropy.coordinates import Angle, SkyCoord # Define a search cone cone_centre = SkyCoord(ra=149, dec=51, unit='deg') cone_radius = Angle(3, unit='deg') # Define a filter-set my_filters = { apiv1.FilterKeys.cone: (cone_centre, cone_radius), apiv1.FilterKeys.ivorn_contains: 'BAT_GRB', apiv1.FilterKeys.role: 'observation', } # See how many results match our filter-set n_results = apiv1.count(filters=my_filters) # List all IVORNs for matching VOEvents, oldest first matching_ivorns = apiv1.list_ivorn( filters=my_filters, order=apiv1.OrderValues.author_datetime, ) # Fetch the raw XML for the first listed VOEvent xml_content = apiv1.packet_xml(matching_ivorns[0])
def test_mock_requests_empty_db(mock_requests): count = apiv1.count() assert count == 0
def test_mock_requests_simpledb(mock_requests, simple_populated_db): dbinf = simple_populated_db count = apiv1.count() assert count == dbinf.n_inserts