Fix test data from static TLE file

1 files changed, 21 insertions, 7 deletions

diff --git a/iss/tests/test_predictions.py b/iss/tests/test_predictions.py
index 27fc667..4c55f5b 100644
--- a/iss/tests/test_predictions.py
+++ b/iss/tests/test_predictions.py
@@ -1,7 +1,11 @@
+from pathlib import Path
 from skyfield.api import Topos
 
 from ..predictions import Predictions
 
+DATA_DIR = Path(__file__).parent / "data"
+STATIONS = DATA_DIR / "stations.txt"
+
 
 def test_get_timescales():
     days = 5
@@ -29,15 +33,25 @@ def test_get_location():
 def test_get_predictions():
     start = 2459117.245895914
 
-    p = Predictions(lat=32.0853, lng=34.7817, tz="Asia/Jerusalem", start=start)
+    p = Predictions(
+        lat=32.0853, lng=34.7817, tz="Asia/Jerusalem", start=start, tle_file=STATIONS
+    )
     preds = p.get_predictions()
-    assert len(preds) == 17
+    assert len(preds) == 16
     assert preds[0] == [
-        "2020-09-24T21:31:09Z",
-        "2020-09-24T21:32:35Z",
-        "2020-09-24T21:34:02Z",
+        "2020-09-24T21:31:31Z",
+        "2020-09-24T21:32:57Z",
+        "2020-09-24T21:34:24Z",
     ]
 
-    p = Predictions(lat=32.0853, lng=34.7817, tz="Asia/Jerusalem", start=start, days=5)
+    p = Predictions(
+        lat=32.0853,
+        lng=34.7817,
+        tz="Asia/Jerusalem",
+        altitude=0,
+        start=start,
+        days=5,
+        tle_file=STATIONS,
+    )
     preds = p.get_predictions()
-    assert len(preds) == 8
+    assert len(preds) == 39