def test_get_query_payload():
nasa_exoplanet_archive = NasaExoplanetArchiveClass()
def mock_run_query(table="ps",
get_query_payload=True,
select="count(*)",
where="disc_facility like '%TESS%'"):
assert table == "ps"
assert get_query_payload
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']
def test_select():
nasa_exoplanet_archive = NasaExoplanetArchiveClass()
def mock_run_query(table="ps", select=["hostname", "pl_name"], where="hostname='Kepler-11'", get_query_payload=True):
assert table == "ps"
assert select == ["hostname", "pl_name"]
assert where == "hostname='Kepler-11'"
assert get_query_payload
payload = PropertyMock()
payload = {'table': 'ps', 'select': 'hostname,pl_name', 'where': "hostname='Kepler-11'", 'format': 'ipac'}
return payload
nasa_exoplanet_archive.query_criteria = mock_run_query
payload = nasa_exoplanet_archive.query_criteria()
assert payload["select"] == "hostname,pl_name"
def test_query_criteria():
nasa_exoplanet_archive = NasaExoplanetArchiveClass()
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 where == "discoverymethod like 'Microlensing' and dec > 0"
result = PropertyMock()
result = {'pl_name': 'TCP J05074264+2447555 b', 'discoverymethod': 'Microlensing', 'dec': [24.7987499] * u.deg}
return result
nasa_exoplanet_archive.query_criteria = mock_run_query
response = nasa_exoplanet_archive.query_criteria()
assert 'TCP J05074264+2447555 b' in response['pl_name']
assert 'Microlensing' in response['discoverymethod']
assert response['dec'] == [24.7987499] * u.deg
def test_api_tables(patch_get, table, query):
NasaExoplanetArchiveMock = NasaExoplanetArchiveClass()
NasaExoplanetArchiveMock._tap_tables = ['list']
data = NasaExoplanetArchiveMock.query_criteria(table, select="*", **query)
assert len(data) > 0
# Check that the units were fixed properly
for col in data.columns:
assert isinstance(data[col], SkyCoord) or not isinstance(data[col].unit, u.UnrecognizedUnit)