Vectorize HEASARC object queries

CHANGES.rst

@@ -8,6 +8,11 @@ New Tools and Services
 Service fixes and enhancements
 ------------------------------
 
+heasarc
+^^^
+
+- Added vectoring for querying objects [#3013]
+
 mpc
 ^^^
 

astroquery/heasarc/core.py

@@ -121,12 +121,12 @@ def query_object_async(self, object_name, mission, *,
                            cache=True, get_query_payload=False,
                            **kwargs):
         """
-        Query around a specific object within a given mission catalog
+        Query around a specific objects within a given mission catalog
 
         Parameters
         ----------
-        object_name : str
-            Object to query around. To set search radius use the 'radius'
+        object_name : str or iterable (list or tuple) of strings
+            Objects to query around. To set search radius use the 'radius'
             parameter.
         mission : str
             Mission table to search from
@@ -139,7 +139,7 @@ def query_object_async(self, object_name, mission, *,
         """
         request_payload = self._args_to_payload(
             mission=mission,
-            entry=object_name,
+            entry=object_name if isinstance(object_name, str) else "; ".join(object_name),
             **kwargs
         )
 

astroquery/heasarc/tests/data/b493ec2a.dat

@@ -12,4 +12,7 @@ Available tables:
 |INTEGRAL-REV3|integral_rev3_prop    |Proposals                                      |
 |INTEGRAL-REV3|integral_rev3_prop_obs|Proposal Information and Observation Parameters|
 |INTEGRAL-REV3|integral_rev3_scw     |SCW - Science Window Data                      |
+|INTEGRAL-REV4|integral_rev4_prop    |Proposals                                      |
+|INTEGRAL-REV4|integral_rev4_prop_obs|Proposal Information and Observation Parameters|
+|INTEGRAL-REV4|integral_rev4_scw     |SCW - Science Window Data                      |
 BatchEnd

astroquery/heasarc/tests/test_heasarc_remote.py

@@ -11,6 +11,7 @@
 
 from astropy.coordinates import SkyCoord
 from astropy import units as u
+from astropy.table import unique
 
 
 @parametrization_local_save_remote
@@ -83,13 +84,25 @@ def test_mission_cols(self):
 
     def test_query_object_async(self):
         mission = 'rosmaster'
-        object_name = '3c273'
+        object_names = ['3c273', 'PSR B0531+21']
 
         heasarc = Heasarc()
-        response = heasarc.query_object_async(object_name, mission=mission)
+        response = heasarc.query_object_async(object_names, mission=mission)
         assert response is not None
         assert isinstance(response, (requests.models.Response, MockResponse))
 
+    def test_query_object(self):
+        mission = "nicermastr"
+        object_names = ("Swift_J1818.0-1607", "SAX_J1808.4-3658")
+
+        heasarc = Heasarc()
+        response = heasarc.query_object(object_name=object_names,
+                                        mission=mission)
+        uresponse = unique(response, keys="NAME")
+        returned_names = [str(i).replace(" ", "") for i in uresponse["NAME"]]
+        assert "Swift_J1818.0-1607" in returned_names
+        assert "SAX_J1808.4-3658" in returned_names
+
     def test_query_region_async(self):
         heasarc = Heasarc()
         mission = 'rosmaster'

astroquery/heasarc/tests/test_heasarc_remote_isdc.py

@@ -162,9 +162,10 @@ def test_mission_list(self):
         with self.isdc_context:
             missions = heasarc.query_mission_list()
 
-        # Assert that there are indeed a large number of tables
-        # Number of tables could change, but should be > 900 (currently 956)
-        assert len(missions) == 5
+        # Assert that there are indeed numerous tables from ISDC
+        # Qunatity of tables could change due to breakdown of the descriptions
+        # Accessing via isdc returns only Integral related tables from HEASARC
+        assert len(missions) >= 5
 
     def test_mission_cols(self):
         heasarc = Heasarc()
@@ -184,10 +185,10 @@ def test_mission_cols(self):
 
     def test_query_object_async(self):
         mission = 'integral_rev3_scw'
-        object_name = '3c273'
+        object_names = ['3c273', "Crab"]
 
         heasarc = Heasarc()
-        response = heasarc.query_object_async(object_name, mission=mission)
+        response = heasarc.query_object_async(object_names, mission=mission)
         assert response is not None
         assert isinstance(response, (requests.models.Response, MockResponse))