diff options
| -rw-r--r-- | iss/predictions.py | 7 | ||||
| -rw-r--r-- | iss/tests/test_predictions.py | 10 |
2 files changed, 15 insertions, 2 deletions
diff --git a/iss/predictions.py b/iss/predictions.py index 1f34a19..ac28cea 100644 --- a/iss/predictions.py +++ b/iss/predictions.py @@ -21,10 +21,13 @@ class Predictions(object): t1 = ts.ut1_jd(t0.ut1 + days) return t0, t1 - def get_predictions(self, location=TEL_AVIV, satellite_name=ISS): + def get_location_topos(self): + return Topos(latitude_degrees=self.lat, longitude_degrees=self.lng) + + def get_predictions(self, location=TEL_AVIV): satellites = self.init_stations() by_name = {sat.name: sat for sat in satellites} - satellite = by_name[satellite_name] + satellite = by_name[self.satellite] t0, t1 = self.get_next_days_timescale() diff --git a/iss/tests/test_predictions.py b/iss/tests/test_predictions.py index 541b2d8..daa2bf4 100644 --- a/iss/tests/test_predictions.py +++ b/iss/tests/test_predictions.py @@ -1,3 +1,5 @@ +from skyfield.api import Topos + from ..predictions import Predictions @@ -7,3 +9,11 @@ def test_get_timescales(): t0, t1 = p.get_next_days_timescale(days=days) delta = t1.ut1 - t0.ut1 assert delta == days + + +def test_get_location(): + p = Predictions(lat=32.0853, lng=34.7817) + tlv = Topos("32.0853 N", "34.7817 E") + topos = p.get_location_topos() + assert topos.latitude.degrees == tlv.latitude.degrees + assert topos.longitude.degrees == tlv.longitude.degrees |