def test_step_unit(self):
with pytest.raises(QueryError):
Ephem.from_mpc('Ceres',
epochs={
'start': Time('2018-10-01'),
'step': 1 * u.year
})
def test_EarthLocation(self):
lowell = EarthLocation.of_site('Lowell Observatory')
eph1 = Ephem.from_miriade("Ceres",
location=lowell,
epochs=Time('2019-01-01'))
eph2 = Ephem.from_miriade("Ceres", epochs=Time('2019-01-01'))
assert abs(eph1['RA'][0] - eph2['RA'][0]) > 0.0001 * u.deg
def test_output_epochs(self):
from astropy.utils.iers import IERSRangeError
# transformed from ut1 to utc
# this test sometimes causes trouble in remote tests, potentially
# because it has to download some data, which might fail;
# if that happens, skip the test
try:
epochs = Time(2437655.5, format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'][0].scale == 'utc'
except IERSRangeError:
pass
# all utc
epochs = Time(2437675.5, format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'][0].scale == 'utc'
# mixed ut1 and utc
# same as above
try:
epochs = Time([2437655.5, 2437675.5], format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'].scale == 'utc'
except IERSRangeError:
pass
def test_minimum_working_columns(self):
"""Avoid assuming any particular columns are returned.
Regression test for #242.
"""
e = Ephem.from_horizons(1, quantities='1') # just RA, Dec
assert len(e) == 1
e = Ephem.from_horizons(1, quantities='19') # just rh and delta-rh
assert len(e) == 1
def test_from_oo():
"""test from_oo method"""
try:
import pyoorb
except ImportError:
return None
orbit = Orbit.from_horizons('Ceres')
horizons_ephem = Ephem.from_horizons('Ceres', location='500')
oo_ephem = Ephem.from_oo(orbit)
u.isclose(horizons_ephem['ra'][0], oo_ephem['ra'][0])
u.isclose(horizons_ephem['dec'][0], oo_ephem['dec'][0])
u.isclose(horizons_ephem['RA*cos(Dec)_rate'][0],
oo_ephem['RA*cos(Dec)_rate'][0])
u.isclose(horizons_ephem['dec_rate'][0], oo_ephem['dec_rate'][0])
u.isclose(horizons_ephem['alpha'][0], oo_ephem['alpha'][0])
u.isclose(horizons_ephem['r'][0], oo_ephem['r'][0])
u.isclose(horizons_ephem['delta'][0], oo_ephem['delta'][0])
u.isclose(horizons_ephem['V'][0], oo_ephem['V'][0])
u.isclose(horizons_ephem['hlon'][0], oo_ephem['hlon'][0])
u.isclose(horizons_ephem['hlat'][0], oo_ephem['hlat'][0])
u.isclose(horizons_ephem['EL'][0], oo_ephem['EL'][0])
# test manual orbit definition lacking units
manorbit = Orbit.from_dict({
'targetname': orbit['targetname'][0],
'a': orbit['a'].value[0],
'e': orbit['e'][0],
'i': orbit['i'].value[0],
'w': orbit['w'].value[0],
'Omega': orbit['Omega'].value[0],
'datetime_jd': orbit['datetime_jd'][0].value,
'M': orbit['M'].value[0],
'H': orbit['H'].value[0],
'G': orbit['G'][0],
'timescale': orbit['timescale'][0]
})
oo_ephem = Ephem.from_oo(manorbit)
u.isclose(horizons_ephem['ra'][0], oo_ephem['ra'][0])
u.isclose(horizons_ephem['dec'][0], oo_ephem['dec'][0])
u.isclose(horizons_ephem['RA*cos(Dec)_rate'][0],
oo_ephem['RA*cos(Dec)_rate'][0])
u.isclose(horizons_ephem['dec_rate'][0], oo_ephem['dec_rate'][0])
u.isclose(horizons_ephem['alpha'][0], oo_ephem['alpha'][0])
u.isclose(horizons_ephem['r'][0], oo_ephem['r'][0])
u.isclose(horizons_ephem['delta'][0], oo_ephem['delta'][0])
u.isclose(horizons_ephem['V'][0], oo_ephem['V'][0])
u.isclose(horizons_ephem['hlon'][0], oo_ephem['hlon'][0])
u.isclose(horizons_ephem['hlat'][0], oo_ephem['hlat'][0])
u.isclose(horizons_ephem['EL'][0], oo_ephem['EL'][0])
def test_single_epoch(self):
# utc
eph1 = Ephem.from_mpc('Ceres', epochs=Time('2018-10-01', scale='utc'))
assert len(eph1.table) == 1
# tdb
with warnings.catch_warnings(record=True) as w:
eph2 = Ephem.from_mpc('Ceres',
epochs=Time('2018-10-01', scale='tdb'))
assert any(
["astroquery.mpc" in str(w[i].message) for i in range(len(w))])
assert all(eph2['epoch'] != eph1['epoch'])
def test_epochTime(self):
# utc
eph1 = Ephem.from_miriade(["Ceres", 'Pallas'],
epochs=Time('2018-10-01', scale='utc'))
assert_quantity_allclose(eph1['epoch'][0].jd, Time('2018-10-01').jd)
# tt
with warnings.catch_warnings(record=True) as w:
eph2 = Ephem.from_miriade(["Ceres", 'Pallas'],
epochs=Time('2018-10-01', scale='tt'))
assert any([
"astroquery.imcce" in str(w[i].message) for i in range(len(w))
])
assert all(eph1['epoch'] != eph2['epoch'])
def test_output_epochs(self):
# transformed from ut1 to utc
epochs = Time(2437655.5, format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'][0].scale == 'utc'
# all utc
epochs = Time(2437675.5, format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'][0].scale == 'utc'
# mixed ut1 and utc
epochs = Time([2437655.5, 2437675.5], format='jd', scale='utc')
a = Ephem.from_horizons(1, epochs=epochs)
assert a['epoch'].scale == 'utc'
def test_start_stop_no_step(self):
with pytest.raises(QueryError):
eph = Ephem.from_mpc('Ceres',
epochs={
'start': Time('2018-10-01'),
'stop': Time('2018-10-31')
})
def test_start_stop_no_step(self):
with pytest.raises(ValueError):
eph = Ephem.from_mpc('Ceres',
epochs={
'start': '2018-10-01',
'stop': '2018-10-31'
})
def test_step_unit(self):
with pytest.raises(ValueError):
eph = Ephem.from_mpc('Ceres',
epochs={
'start': '2018-10-01',
'step': '1yr'
})
def test_start_stop_step(self):
# utc
epochs = dict(start=Time('2018-10-01'),
stop=Time('2018-10-31'),
step=1 * u.d)
eph1 = Ephem.from_mpc('Ceres', epochs=epochs)
assert len(eph1.table) == 31
# tt
epochs = dict(start=Time('2018-10-01', scale='tt'),
stop=Time('2018-10-31', scale='tt'),
step=1 * u.d)
with warnings.catch_warnings(record=True) as w:
eph2 = Ephem.from_mpc('Ceres', epochs=epochs)
assert any(
["astroquery.mpc" in str(w[i].message) for i in range(len(w))])
assert all(eph2['epoch'] != eph1['epoch'])
def test_daterange_step(self):
epochs = {
'start': Time('2018-01-02', format='iso'),
'stop': Time('2018-01-05', format='iso'),
'step': 6 * u.hour
}
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 13
def test_ra_dec_format(self):
epochs = dict(start='2018-10-01', step='1d', number=31)
ra_format = {'sep': ':', 'unit': 'hourangle', 'precision': 1}
dec_format = {'sep': ':', 'precision': 1}
eph = Ephem.from_mpc('Ceres', epochs=epochs, ra_format=ra_format,
dec_format=dec_format)
assert isinstance(eph['RA'][0], str)
assert isinstance(eph['Dec'][0], str)
def test_timescale(self):
# test same timescale
time = Time('2020-01-01 00:00', format='iso', scale='utc')
a = Ephem.from_horizons(1, epochs=time)
assert a['epoch'].scale == 'utc'
assert a['epoch'].utc.jd == 2458849.5
# test different timescale (and check for warning)
time = Time('2020-01-01 00:00', format='iso', scale='tdb')
with warnings.catch_warnings(record=True) as w:
a = Ephem.from_horizons(1, epochs=time)
assert any([
"astroquery.jplhorizons" in str(w[i].message)
for i in range(len(w))
])
assert a['epoch'].scale == 'utc'
assert_allclose(a['epoch'][0].utc.jd, 2458849.49919926)
def test_daterange_number(self):
epochs = {
'start': Time('2018-01-02', format='iso'),
'stop': Time('2018-01-05', format='iso'),
'number': 10
}
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 10
def test_epochrange_step(self):
eph = Ephem.from_miriade(
["Ceres", 'Pallas'],
epochs={
'start': Time('2018-10-01'),
'stop': Time('2018-10-10'),
'step': 0.5 * u.d
})
assert len(eph.table) == 38
def test_start_stop_jd(self):
epochs = {
'start': Time(2458396.5, format='jd'),
'stop': Time(2458397.5, format='jd'),
'step': 1 * u.d
}
eph = Ephem.from_mpc('Ceres', epochs=epochs)
assert eph['Date'][0] == '2018-10-05 00:00:00.000'
assert eph['Date'][1] == '2018-10-06 00:00:00.000'
def test_from_horizons():
""" test from_horizons method"""
# current epoch
now = Time.now()
data = Ephem.from_horizons('Ceres')
assert_allclose(data.datetime_jd, now.jd * u.d)
# date range - astropy.time.Time objects
epochs = {
'start': Time('2018-01-02', format='iso'),
'stop': Time('2018-01-05', format='iso'),
'step': '6h'
}
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 13
# date range - strings
epochs = {'start': '2018-01-02', 'stop': '2018-01-05', 'step': '6h'}
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 13
# astropy.time.Time object with list of Dates
epochs = Time(['2018-01-01', '2018-01-02'])
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 2
# query two objects
data = Ephem.from_horizons(['Ceres', 'Pallas'])
assert len(data.table) == 2
# test bib service
bib.track()
data = Ephem.from_horizons(['Ceres', 'Pallas'])
assert 'sbpy.data.Ephem' in bib.to_text()
def test_epochrange_number(self):
# utc
eph1 = Ephem.from_miriade(["Ceres", 'Pallas'],
epochs={
'start': Time('2018-10-01'),
'stop': Time('2018-11-01'),
'number': 10
})
assert len(eph1.table) == 20
# tdb
with warnings.catch_warnings(record=True) as w:
eph2 = Ephem.from_miriade(
["Ceres", 'Pallas'],
epochs={
'start': Time('2018-10-01', scale='tdb'),
'stop': Time('2018-11-01', scale='tdb'),
'number': 10
})
assert any([
"astroquery.imcce" in str(w[i].message) for i in range(len(w))
])
assert all(eph1['epoch'] != eph2['epoch'])
def test_queryfail(self):
with pytest.raises(QueryError):
Ephem.from_horizons('target does not exist')
def test_bib(self):
bib.track()
Ephem.from_horizons(['Ceres', 'Pallas'])
assert 'sbpy.data.ephem.Ephem.from_horizons' in bib.to_text()
def test_Earthlocation(self):
lowell = EarthLocation.of_site('Lowell Observatory')
data = Ephem.from_horizons(1,
epochs=Time('2018-01-01', format='iso'),
location=lowell)
assert len(data.table) == 1
def test_multiple_objects(self):
data = Ephem.from_horizons(['Ceres', 'Pallas'])
assert len(data.table) == 2
def test_Time_list(self):
epochs = Time(['2018-01-01', '2018-01-02'])
data = Ephem.from_horizons('Ceres', epochs=epochs)
assert len(data.table) == 2
def test_bib(self):
bib.track()
orbit = Orbit.from_horizons('Ceres')
oo_ephem = Ephem.from_oo(orbit, scope='basic')
assert 'sbpy.data.ephem.Ephem.from_oo' in bib.to_text()
def test_timescale(self):
orbit = Orbit.from_horizons('Ceres')
oo_ephem = Ephem.from_oo(orbit, scope='basic')
assert oo_ephem['epoch'].scale == 'tai'
def test_basic(self):
orbit = Orbit.from_horizons('Ceres')
oo_ephem = Ephem.from_oo(orbit, scope='basic')
assert 'dec_rate' not in oo_ephem.field_names
def test_bib(self):
bib.track()
data = Ephem.from_miriade(['Ceres', 'Pallas'])
assert 'sbpy.data.ephem.Ephem.from_miriade' in bib.to_text()
def test_iaulocation(self):
eph1 = Ephem.from_miriade("Ceres",
location='G37',
epochs=Time('2019-01-01'))
eph2 = Ephem.from_miriade("Ceres", epochs=Time('2019-01-01'))
assert abs(eph1['RA'][0] - eph2['RA'][0]) > 0.00001 * u.deg
