adding moving tesscut tests

Author: Clara Brasseur

astroquery/mast/tests/test_mast.py

@@ -589,6 +589,15 @@ def test_tesscut_get_sector(patch_post):
     assert sector_table['camera'][0] == 1
     assert sector_table['ccd'][0] == 3
 
+    # Exercising the search by moving target
+    sector_table = mast.Tesscut.get_sectors(moving_target="Ceres")
+    assert isinstance(sector_table, Table)
+    assert len(sector_table) == 1
+    assert sector_table['sectorName'][0] == "tess-s0001-1-3"
+    assert sector_table['sector'][0] == 1
+    assert sector_table['camera'][0] == 1
+    assert sector_table['ccd'][0] == 3
+
 
 def test_tesscut_download_cutouts(patch_post, tmpdir):
 
@@ -616,6 +625,13 @@ def test_tesscut_download_cutouts(patch_post, tmpdir):
     assert manifest["Local Path"][0][-4:] == "fits"
     assert os.path.isfile(manifest[0]['Local Path'])
 
+    # Exercising the search by moving target
+    manifest = mast.Tesscut.download_cutouts(moving_target="Eleonora", size=5, path=str(tmpdir))
+    assert isinstance(manifest, Table)
+    assert len(manifest) == 1
+    assert manifest["Local Path"][0][-4:] == "fits"
+    assert os.path.isfile(manifest[0]['Local Path'])
+
 
 def test_tesscut_get_cutouts(patch_post, tmpdir):
 
@@ -631,6 +647,13 @@ def test_tesscut_get_cutouts(patch_post, tmpdir):
     assert len(cutout_hdus_list) == 1
     assert isinstance(cutout_hdus_list[0], fits.HDUList)
 
+    # Exercising the search by object name
+    cutout_hdus_list = mast.Tesscut.get_cutouts(moving_target="Eleonora", size=5)
+    assert isinstance(cutout_hdus_list, list)
+    assert len(cutout_hdus_list) == 1
+    assert isinstance(cutout_hdus_list[0], fits.HDUList)
+
+
 ######################
 # ZcutClass tests #
 ######################

astroquery/mast/tests/test_mast_remote.py

@@ -14,6 +14,8 @@
 from astroquery.exceptions import NoResultsWarning
 from astroquery import mast
 
+from ...exceptions import RemoteServiceError, MaxResultsWarning
+
 
 OBSID = '1647157'
 
@@ -195,10 +197,13 @@ def test_observations_query_criteria_count(self):
 
     # product functions
     def test_observations_get_product_list_async(self):
-        responses = mast.Observations.get_product_list_async('2003738726')
+ 
+        test_obs = mast.Observations.query_criteria(filters=["NUV","FUV"],objectname="M101")
+
+        responses = mast.Observations.get_product_list_async(test_obs[0]["obsid"])
         assert isinstance(responses, list)
 
-        responses = mast.Observations.get_product_list_async('2003738726,3000007760')
+        responses = mast.Observations.get_product_list_async(test_obs[2:3])
         assert isinstance(responses, list)
 
         observations = mast.Observations.query_object("M8", radius=".02 deg")
@@ -242,8 +247,12 @@ def test_observations_get_product_list(self):
         result = mast.Observations.get_product_list(observations[obsLocs])
         obs_collection = np.unique(list(result['obs_collection']))
         assert isinstance(result, Table)
+<<<<<<< HEAD
         assert len(obs_collection) == 1
         assert obs_collection[0] == 'IUE'
+=======
+        assert len(result) > 10
+>>>>>>> adding moving tesscut tests
 
     def test_observations_filter_products(self):
         observations = mast.Observations.query_object("M8", radius=".04 deg")
@@ -269,7 +278,7 @@ def test_observations_download_products(self, tmpdir):
                 assert os.path.isfile(row['Local Path'])
 
         # just get the curl script
-        result = mast.Observations.download_products(test_obs_id,
+        result = mast.Observations.download_products(test_obs[0]["obsid"],
                                                      download_dir=str(tmpdir),
                                                      curl_flag=True,
                                                      productType=["SCIENCE"],
@@ -278,17 +287,18 @@ def test_observations_download_products(self, tmpdir):
         assert os.path.isfile(result['Local Path'][0])
 
         # check for row input
-        result1 = mast.Observations.get_product_list(test_obs_id)
+        result1 = mast.Observations.get_product_list(test_obs[0]["obsid"])
         result2 = mast.Observations.download_products(result1[0])
         assert isinstance(result2, Table)
         assert os.path.isfile(result2['Local Path'][0])
         assert len(result2) == 1
 
     def test_observations_download_file(self, tmpdir):
         test_obs_id = OBSID
+        test_obs = mast.Observations.query_criteria(filters=["NUV","FUV"],objectname="M101")
 
         # pull a single data product
-        products = mast.Observations.get_product_list(test_obs_id)
+        products = mast.Observations.get_product_list(test_obs[0]["obsid"])
         uri = products['dataURI'][0]
 
         # download it
@@ -364,6 +374,10 @@ def test_catalogs_query_region(self):
                                             catalog="HSC",
                                             magtype=2)
         row = np.where(result['MatchID'] == '78095437')
+
+        with pytest.warns(MaxResultsWarning):
+            result = mast.Catalogs.query_region("322.49324 12.16683", catalog="HSC", magtype=2)
+
         assert isinstance(result, Table)
         assert result[row]['NumImages'] == 1
         assert result[row]['TargetName'] == 'M15'
@@ -374,6 +388,10 @@ def test_catalogs_query_region(self):
                                             version=2,
                                             magtype=2)
         row = np.where(result['MatchID'] == '82368728')
+
+        with pytest.warns(MaxResultsWarning):
+            result = mast.Catalogs.query_region("322.49324 12.16683", catalog="HSC",
+                                                version=2, magtype=2)
         assert isinstance(result, Table)
         assert result[row]['NumImages'] == 11
         assert result[row]['TargetName'] == 'NGC7078'
@@ -666,13 +684,20 @@ def test_tesscut_get_sectors(self):
         assert sector_table['ccd'][0] > 0
 
         # This should always return no results
+
         with pytest.warns(NoResultsWarning):
             coord = SkyCoord(90, -66.5, unit="deg")
             sector_table = mast.Tesscut.get_sectors(coordinates=coord,
                                                     radius=0)
             assert isinstance(sector_table, Table)
             assert len(sector_table) == 0
 
+        coord = SkyCoord(90, -66.5, unit="deg")
+        with pytest.warns(NoResultsWarning):
+            sector_table = mast.Tesscut.get_sectors(coordinates=coord, radius=0)
+        assert isinstance(sector_table, Table)
+        assert len(sector_table) == 0
+
         sector_table = mast.Tesscut.get_sectors(objectname="M104")
         assert isinstance(sector_table, Table)
         assert len(sector_table) >= 1
@@ -681,6 +706,15 @@ def test_tesscut_get_sectors(self):
         assert sector_table['camera'][0] > 0
         assert sector_table['ccd'][0] > 0
 
+        sector_table = mast.Tesscut.get_sectors(moving_target="Stichius")
+        assert isinstance(sector_table, Table)
+        assert len(sector_table) >= 1
+        assert sector_table['sectorName'][0] == "tess-s0001-1-1"
+        assert sector_table['sector'][0] == 1
+        assert sector_table['camera'][0] == 1
+        assert sector_table['ccd'][0] == 1
+
+
     def test_tesscut_download_cutouts(self, tmpdir):
 
         coord = SkyCoord(349.62609, -47.12424, unit="deg")
@@ -720,6 +754,14 @@ def test_tesscut_download_cutouts(self, tmpdir):
         for row in manifest:
             assert os.path.isfile(row['Local Path'])
 
+        manifest = mast.Tesscut.download_cutouts(moving_target="Eleonora", sector=6, size=5, path=str(tmpdir))
+        assert isinstance(manifest, Table)
+        assert len(manifest) == 1
+        assert manifest["Local Path"][0][-4:] == "fits"
+        for row in manifest:
+            assert os.path.isfile(row['Local Path'])
+        
+
     def test_tesscut_get_cutouts(self, tmpdir):
 
         coord = SkyCoord(107.18696, -70.50919, unit="deg")
@@ -746,9 +788,15 @@ def test_tesscut_get_cutouts(self, tmpdir):
         assert len(cutout_hdus_list) >= 1
         assert isinstance(cutout_hdus_list[0], fits.HDUList)
 
-    ######################
+        cutout_hdus_list = mast.Tesscut.get_cutouts(moving_target="Eleonora", sector=6, size=5)
+        assert isinstance(cutout_hdus_list, list)
+        assert len(cutout_hdus_list) == 1
+        assert isinstance(cutout_hdus_list[0], fits.HDUList)
+
+        
+    ###################
     # ZcutClass tests #
-    ######################
+    ###################
     def test_zcut_get_surveys(self):
 
         coord = SkyCoord(189.49206, 62.20615, unit="deg")
@@ -766,6 +814,12 @@ def test_zcut_get_surveys(self):
             assert isinstance(survey_list, list)
             assert len(survey_list) == 0
 
+        coord = SkyCoord(57.10523, -30.08085, unit="deg")
+        with pytest.warns(NoResultsWarning):
+            survey_list = mast.Zcut.get_surveys(coordinates=coord, radius=0)
+        assert isinstance(survey_list, list)
+        assert len(survey_list) == 0
+
     def test_zcut_download_cutouts(self, tmpdir):
 
         coord = SkyCoord(34.47320, -5.24271, unit="deg")
@@ -777,7 +831,7 @@ def test_zcut_download_cutouts(self, tmpdir):
         for row in cutout_table:
             assert os.path.isfile(cutout_table[0]['Local Path'])
 
-        coord = SkyCoord(189.49206, 62.20615, unit="deg")
+        coord = SkyCoord(189.28065571, 62.17415175, unit="deg")
 
         cutout_table = mast.Zcut.download_cutouts(coordinates=coord, size=[200, 300], path=str(tmpdir))
         assert isinstance(cutout_table, Table)
@@ -809,6 +863,13 @@ def test_zcut_download_cutouts(self, tmpdir):
             assert isinstance(cutout_table, Table)
             assert len(cutout_table) == 0
 
+        cutout_table = mast.Zcut.download_cutouts(coordinates=coord, survey='goods_north', cutout_format="jpg", path=str(tmpdir))
+        assert isinstance(cutout_table, Table)
+        assert len(cutout_table) == 4
+        assert cutout_table["Local Path"][0][-4:] == ".jpg"
+        for row in cutout_table:
+            assert os.path.isfile(cutout_table[0]['Local Path'])
+
         cutout_table = mast.Zcut.download_cutouts(coordinates=coord, cutout_format="jpg", path=str(tmpdir), stretch='asinh', invert=True)
         assert isinstance(cutout_table, Table)
         assert len(cutout_table) >= 1
@@ -818,7 +879,7 @@ def test_zcut_download_cutouts(self, tmpdir):
 
     def test_zcut_get_cutouts(self):
 
-        coord = SkyCoord(189.49206, 62.20615, unit="deg")
+        coord = SkyCoord(189.28065571, 62.17415175, unit="deg")
 
         cutout_list = mast.Zcut.get_cutouts(coordinates=coord)
         assert isinstance(cutout_list, list)
@@ -836,3 +897,9 @@ def test_zcut_get_cutouts(self):
                                                 survey='candels_gn_30mas')
             assert isinstance(cutout_list, list)
             assert len(cutout_list) == 0
+
+        cutout_list = mast.Zcut.get_cutouts(coordinates=coord, survey='3dhst_goods-n')
+        assert isinstance(cutout_list, list)
+        assert len(cutout_list) == 1
+        assert isinstance(cutout_list[0], fits.HDUList)
+