def test_missing_criterion_super_wasp():
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_criteria("superwasptimeseries")
assert "Queries against the SuperWASP Time Series table require" in str(
error)
NasaExoplanetArchive.query_criteria("superwasptimeseries",
sourceid="1SWASP J191645.46+474912.3")
def test_missing_criterion_kepler():
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_criteria("keplertimeseries",
where="kepid=8561063")
assert "Queries against the Kepler Time Series table require" in str(error)
NasaExoplanetArchive.query_criteria("keplertimeseries",
kepid=8561063,
quarter=14)
def test_missing_criterion_kelt():
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_criteria("kelttimeseries")
assert "Queries against the KELT Time Series table require" in str(error)
NasaExoplanetArchive.query_criteria(
"kelttimeseries",
where="kelt_sourceid='KELT_N02_lc_012738_V01_east'",
kelt_field="N02")
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)
assert len(table1) == 1
assert table1["hostname"] == "HD 209458"
table2 = NasaExoplanetArchive.query_region("pscomppars", coords, radius)
_compare_tables(table1, table2)
def test_select():
payload = NasaExoplanetArchive.query_criteria(
"ps",
select=["hostname", "pl_name"],
where="hostname='Kepler-11'",
get_query_payload=True,
)
assert payload["select"] == "hostname,pl_name"
table1 = NasaExoplanetArchive.query_criteria(
"ps", select=["hostname", "pl_name"], where="hostname='Kepler-11'")
table2 = NasaExoplanetArchive.query_criteria("ps",
select="hostname,pl_name",
where="hostname='Kepler-11'")
_compare_tables(table1, table2)
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)
table1 = NasaExoplanetArchive.query_object("HAT-P-11 b", format="csv")
table2 = NasaExoplanetArchive.query_object("HAT-P-11 b", format="bar")
_compare_tables(table1, table2)
table1 = NasaExoplanetArchive.query_object("HAT-P-11 b", format="xml")
table2 = NasaExoplanetArchive.query_object("HAT-P-11 b", format="table")
_compare_tables(table1, table2)
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_object("HAT-P-11 b", format="json")
assert "json" in str(error)
def test_warnings():
with pytest.warns(NoResultsWarning):
NasaExoplanetArchive.query_criteria("ps",
where="hostname='not a host'")
with pytest.warns(InputWarning):
NasaExoplanetArchive.query_object("HAT-P-11 b", where="nothing")
with pytest.warns(InputWarning):
NasaExoplanetArchive.query_object(object_name="K2-18 b",
table="pscomppars",
where="nothing")
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_object("HAT-P-11 b", table="cumulative")
assert "Invalid table 'cumulative'" in str(error)
def test_table_errors():
with pytest.raises(InvalidTableError) as error:
NasaExoplanetArchive.query_object("K2-18 b", table="exoplanets")
assert "exoplanets" in str(error)
with pytest.raises(InvalidTableError) as error:
NasaExoplanetArchive.query_object("K2-18 b", table="exomultpars")
assert "exomultpars" in str(error)
with pytest.raises(InvalidTableError) as error:
NasaExoplanetArchive.query_object("K2-18 b", table="compositepars")
assert "compositepars" in str(error)
def test_request_to_sql():
payload_dict = NasaExoplanetArchive.query_region(table="ps",
coordinates=SkyCoord(
ra=172.56 * u.deg,
dec=7.59 * u.deg),
radius=1.0 * u.deg,
get_query_payload=True)
sql_str = NasaExoplanetArchive._request_to_sql(payload_dict)
assert sql_str == "select * from ps where contains(point('icrs',ra,dec),circle('icrs',172.56,7.59,1.0 degree))=1"
payload_dict = NasaExoplanetArchive.query_criteria(
table="ps",
where="hostname like 'Kepler%'",
order="hostname",
get_query_payload=True)
assert "order by" in NasaExoplanetArchive._request_to_sql(payload_dict)
payload_dict = NasaExoplanetArchive.query_criteria(
table="cumulative",
where="pl_hostname like 'Kepler%'",
order="pl_hostname",
get_query_payload=True)
assert "pl_hostname or pl_name" in NasaExoplanetArchive._request_to_sql(
payload_dict)
def test_invalid_query_exoplanets():
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_criteria("ps", where="hostname=Kepler-11")
assert "kepler" in str(error).lower()
def test_invalid_column():
with pytest.raises(InvalidQueryError) as error:
NasaExoplanetArchive.query_criteria("ps", select="not_a_column")
assert "not_a_column" in str(error).lower()
def test_query_aliases():
name = "bet Pic"
aliases = NasaExoplanetArchive.query_aliases(name)
assert len(aliases) > 10
assert "HD 39060" in aliases
def test_invalid_table():
with pytest.raises(InvalidTableError) as error:
NasaExoplanetArchive.query_criteria("not_a_table")
assert "not_a_table" in str(error)