modify test

Author: Johannes-Sahlmann

pystrometry/tests/test_get_ephemeris.py

@@ -1,6 +1,10 @@
+import os
+
 from astropy.table import Table
 from astropy.time import Time
+
 from ..pystrometry import get_ephemeris
+from ..pystrometry import ephemeris_dir
 
 def test_get_ephemeris_valid_input():
     # Define valid test inputs
@@ -10,6 +14,11 @@ def test_get_ephemeris_valid_input():
     stop_time = Time('2024-01-10T00:00:00')
     step_size = '1d'
 
+    print(f"\n{ephemeris_dir}\n")
+    horizons_file_seed = '{}_{}_{}_{}_{}'.format(center, target, start_time, stop_time, step_size)
+    out_file = os.path.join(ephemeris_dir, horizons_file_seed + '.txt')
+    assert os.path.isfile(out_file), f"Expected output file {out_file} does not exist."
+
     # Call the function
     result = get_ephemeris(center=center, target=target, start_time=start_time, stop_time=stop_time, step_size=step_size)