Fix timescale declaration in query_neocc

B612-Asteroid-Institute · Jan 30, 2025 · 24e3fcf · 24e3fcf
1 parent 1bb32ff
commit 24e3fcf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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()))