inverse tags and make tests works

FusRoman · Nov 27, 2023 · 4ff7a4f · 4ff7a4f
1 parent 804dc98
commit 4ff7a4f
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 9 deletions.
diff --git a/bin/utils_cli.py b/bin/utils_cli.py
@@ -442,11 +442,11 @@ def assig_tags(orb_df, traj_orb_df, start_tags):
     ... })
     >>> new_orb, new_traj = assig_tags(orb, traj, 0)
 
-    >>> orb_test = pd.read_parquet("fink_fat/test/utils_cli_test_orb.parquet")
-    >>> traj_test = pd.read_parquet("fink_fat/test/utils_cli_test_traj.parquet")
+    >>> orb_test = pd.read_parquet("fink_fat/test/utils_cli_test_orb.parquet").reset_index(drop=True)
+    >>> traj_test = pd.read_parquet("fink_fat/test/utils_cli_test_traj.parquet").reset_index(drop=True)
 
-    >>> assert_frame_equal(orb_test, new_orb)
-    >>> assert_frame_equal(traj_test, new_traj)
+    >>> assert_frame_equal(orb_test, new_orb.reset_index(drop=True))
+    >>> assert_frame_equal(traj_test, new_traj.reset_index(drop=True))
     """
     orb_df = orb_df.sort_values("ref_epoch")
 

diff --git a/fink_fat/others/id_tags.py b/fink_fat/others/id_tags.py
@@ -28,11 +28,11 @@ def int_to_tags(traj_id, jd):
     Examples
     --------
     >>> int_to_tags(15, 2460135.98)
-    'FF10072023aaaaaap'
+    'FF20230710aaaaaap'
     >>> int_to_tags(27, 2460135.98)
-    'FF10072023aaaaabb'
+    'FF20230710aaaaabb'
     >>> int_to_tags(652, 2460135.98)
-    'FF10072023aaaaazc'
+    'FF20230710aaaaazc'
     """
     res_tag = ""
     for _ in range(7):
@@ -43,7 +43,7 @@ def int_to_tags(traj_id, jd):
         traj_id = q
 
     discovery = Time(jd, format="jd").datetime
-    return "FF{:02d}{:02d}{}{}".format(
+    return "FF{:04d}{:02d}{:02d}{}".format(
         discovery.year, discovery.month, discovery.day, res_tag[::-1]
     )
 
@@ -70,7 +70,7 @@ def generate_tags(begin, end, jd):
     Examples
     --------
     >>> generate_tags(3, 6, [2460135.42, 2460137.57, 2460148.72])
-    ['FF09072023aaaaaad', 'FF12072023aaaaaae', 'FF23072023aaaaaaf']
+    ['FF20230709aaaaaad', 'FF20230712aaaaaae', 'FF20230723aaaaaaf']
     """
     return [int_to_tags(i, date) for date, i in zip(jd, np.arange(begin, end))]
 

diff --git a/fink_fat/test/cli_test/fink_fat_out_test/mpc/orbital.parquet b/fink_fat/test/cli_test/fink_fat_out_test/mpc/orbital.parquet
diff --git a/fink_fat/test/cli_test/fink_fat_out_test/mpc/trajectory_orb.parquet b/fink_fat/test/cli_test/fink_fat_out_test/mpc/trajectory_orb.parquet
diff --git a/fink_fat/test/utils_cli_test_orb.parquet b/fink_fat/test/utils_cli_test_orb.parquet
diff --git a/fink_fat/test/utils_cli_test_traj.parquet b/fink_fat/test/utils_cli_test_traj.parquet