Add a few more tests

Author: rickynilsson

astroquery/nasa_exoplanet_archive/tests/test_nasa_exoplanet_archive.py

@@ -250,7 +250,7 @@ def test_query_criteria():
 
     def mock_run_query(table="ps", select='pl_name,discoverymethod,dec', where="discoverymethod like 'Microlensing' and dec > 0"):
         assert table == "ps"
-        assert select == 'pl_name,discoverymethod,dec'
+        assert select == "pl_name,discoverymethod,dec"
         assert where == "discoverymethod like 'Microlensing' and dec > 0"
         result = PropertyMock()
         result = {'pl_name': 'TCP J05074264+2447555 b', 'discoverymethod': 'Microlensing', 'dec': [24.7987499] * u.deg}
@@ -260,3 +260,25 @@ def mock_run_query(table="ps", select='pl_name,discoverymethod,dec', where="disc
     assert 'TCP J05074264+2447555 b' in response['pl_name']
     assert 'Microlensing' in response['discoverymethod']
     assert response['dec'] == [24.7987499] * u.deg
+
+
+@patch('astroquery.nasa_exoplanet_archive.core.get_access_url',
+       Mock(side_effect=lambda x: 'https://some.url'))
+@pytest.mark.skipif(not pyvo_OK, reason='not pyvo_OK')
+def test_get_query_payload():
+    nasa_exoplanet_archive = NasaExoplanetArchive()
+
+    def mock_run_query(table="ps", get_query_payload=True, select="count(*)", where="disc_facility like '%TESS%'"):
+        assert table == "ps"
+        assert get_query_payload == True
+        assert select == "count(*)"
+        assert where == "disc_facility like '%TESS%'"
+        result = PropertyMock()
+        result = {'table': 'ps', 'select': 'count(*)', 'where': "disc_facility like '%TESS%'", 'format': 'ipac'}
+        return result
+    nasa_exoplanet_archive.query_criteria = mock_run_query
+    response = nasa_exoplanet_archive.query_criteria()
+    assert 'ps' in response['table']
+    assert 'count(*)' in response['select']
+    assert "disc_facility like '%TESS%'" in response['where']
+    

astroquery/nasa_exoplanet_archive/tests/test_nasa_exoplanet_archive_remote.py

@@ -98,7 +98,7 @@ def test_select():
 
 
 @pytest.mark.remote_data
-def test_warnings():  # removed patch_get for now
+def test_warnings():
     with pytest.warns(NoResultsWarning):
         NasaExoplanetArchive.query_criteria("ps", where="hostname='not a host'")
 
@@ -142,7 +142,7 @@ def test_request_to_sql():
 
 
 @pytest.mark.remote_data
-def test_query_region():  # removed patch_get for now
+def test_query_region():
     coords = SkyCoord(ra=330.79488 * u.deg, dec=18.8843 * u.deg)
     radius = 0.001
     table1 = NasaExoplanetArchive.query_region("pscomppars", coords, radius * u.deg)
@@ -154,7 +154,15 @@ def test_query_region():  # removed patch_get for now
 
 
 @pytest.mark.remote_data
-def test_format():  # removed patch_get for now
+def test_query_aliases():
+    name = "bet Pic"
+    aliases = NasaExoplanetArchive.query_aliases(name)
+    assert len(aliases) == 12
+    assert "HD 39060" in aliases
+
+
+@pytest.mark.remote_data
+def test_format():
     table1 = NasaExoplanetArchive.query_object("HAT-P-11 b")
     table2 = NasaExoplanetArchive.query_object("HAT-P-11 b", format="votable")
     _compare_tables(table1, table2)