Merge pull request #2504 from jaymedina/mast-convert-lists

converting input lists to np arrays

Author: bsipocz

CHANGES.rst

@@ -16,6 +16,11 @@ hsa
 
 - New module to access ESA Herschel mission. [#2122]
 
+mast
+^^^
+
+- Fixed ``Observations.get_product_list`` to handle input lists of obsids. [#2504]
+
 
 Service fixes and enhancements
 ------------------------------

astroquery/mast/observations.py

@@ -440,9 +440,11 @@ def get_product_list_async(self, observations):
             observations = np.array([observations])
         if isinstance(observations, Table):
             observations = observations['obsid']
+        if isinstance(observations, list):
+            observations = np.array(observations)
 
         observations = observations[observations != ""]
-        if len(observations) == 0:
+        if observations.size == 0:
             raise InvalidQueryError("Observation list is empty, no associated products.")
 
         service = self._caom_products

astroquery/mast/tests/test_mast.py

@@ -441,6 +441,10 @@ def test_observations_get_product_list(patch_post):
     result = mast.Observations.get_product_list(observations[0:4])
     assert isinstance(result, Table)
 
+    in_obsids = ['83229830', '1829332', '26074149', '24556715']
+    result = mast.Observations.get_product_list(in_obsids)
+    assert isinstance(result, Table)
+
 
 def test_observations_filter_products(patch_post):
     products = mast.Observations.get_product_list('2003738726')