From 24e3fcf442717ef4c609b037fa2253917200c479 Mon Sep 17 00:00:00 2001 From: Joachim Moeyens Date: Thu, 30 Jan 2025 14:08:56 -0800 Subject: [PATCH] Fix timescale declaration in query_neocc --- src/adam_core/orbits/query/neocc.py | 2 +- src/adam_core/orbits/query/tests/test_neocc.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/src/adam_core/orbits/query/neocc.py b/src/adam_core/orbits/query/neocc.py index d4d074ef..9b250772 100644 --- a/src/adam_core/orbits/query/neocc.py +++ b/src/adam_core/orbits/query/neocc.py @@ -219,7 +219,7 @@ def query_neocc( time_scale = data["time_system"] if time_scale == "TDT": - time_scale = "TT" + time_scale = "tt" else: raise ValueError(f"Unsupported time scale: {time_scale}") diff --git a/src/adam_core/orbits/query/tests/test_neocc.py b/src/adam_core/orbits/query/tests/test_neocc.py index ce59ed35..83ea55e6 100644 --- a/src/adam_core/orbits/query/tests/test_neocc.py +++ b/src/adam_core/orbits/query/tests/test_neocc.py @@ -158,6 +158,9 @@ def mock_get(url, params): assert orbits.object_id[0].as_py() == "2024YR4" assert orbits.orbit_id[1].as_py() == "2022OB5" assert orbits.object_id[1].as_py() == "2022OB5" + assert orbits.coordinates.time.days[0].as_py() == 60800 + assert orbits.coordinates.time.days[1].as_py() == 60600 + assert orbits.coordinates.time.scale == "tt" assert np.all(~np.isnan(orbits.coordinates.values)) assert np.all(~np.isnan(orbits.coordinates.covariance.to_matrix())) @@ -187,6 +190,9 @@ def mock_get(url, params): assert orbits.object_id[0].as_py() == "2024YR4" assert orbits.orbit_id[1].as_py() == "2022OB5" assert orbits.object_id[1].as_py() == "2022OB5" + assert orbits.coordinates.time.days[0].as_py() == 60688 + assert orbits.coordinates.time.days[1].as_py() == 59792 + assert orbits.coordinates.time.scale == "tt" assert np.all(~np.isnan(orbits.coordinates.values)) assert np.all(~np.isnan(orbits.coordinates.covariance.to_matrix()))