Bug fix: Convert test orbit ID from pyarrow StringScalar to a python string when creating a test orbit directory path

Author: moeyensj

thor/main.py

@@ -47,7 +47,9 @@ def initialize_test_orbit(
     Initialize the test orbit by saving it to disk if a working directory is provided.
     """
     if working_dir is not None:
-        test_orbit_directory = pathlib.Path(working_dir, "inputs", test_orbit.orbit_id)
+        test_orbit_directory = pathlib.Path(
+            working_dir, "inputs", test_orbit.orbit_id[0].as_py()
+        )
         test_orbit_directory.mkdir(parents=True, exist_ok=True)
         test_orbit_path = os.path.join(test_orbit_directory, "test_orbit.parquet")
         test_orbit.to_parquet(test_orbit_path)
@@ -113,7 +115,7 @@ def link_test_orbit(
     if working_dir is not None:
         working_dir_path = pathlib.Path(working_dir)
         logger.info(f"Using working directory: {working_dir}")
-        test_orbit_directory = pathlib.Path(working_dir_path, test_orbit.orbit_id)
+        test_orbit_directory = pathlib.Path(working_dir, test_orbit.orbit_id[0].as_py())
         test_orbit_directory.mkdir(parents=True, exist_ok=True)
         inputs_dir = pathlib.Path(working_dir_path, "inputs")
         inputs_dir.mkdir(parents=True, exist_ok=True)

thor/tests/test_main.py

@@ -232,12 +232,7 @@ def test_link_test_orbit(
     else:
         integration_config.max_processes = 1
 
-    (
-        test_orbit,
-        observations,
-        obs_ids_expected,
-        integration_config,
-    ) = setup_test_data(
+    (test_orbit, observations, obs_ids_expected, integration_config,) = setup_test_data(
         object_id, orbits, observations, integration_config, max_arc_length=14
     )
 
@@ -264,12 +259,7 @@ def test_benchmark_link_test_orbit(
     else:
         integration_config.max_processes = 1
 
-    (
-        test_orbit,
-        observations,
-        obs_ids_expected,
-        integration_config,
-    ) = setup_test_data(
+    (test_orbit, observations, obs_ids_expected, integration_config,) = setup_test_data(
         object_id, orbits, observations, integration_config, max_arc_length=14
     )