Fix name_ fields for integration parse

tatsuya-ishihara · Oct 18, 2024 · 7fd1d6d · 7fd1d6d
1 parent 4a60678
commit 7fd1d6d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/integration/integration_test_parse_logs.py b/tests/integration/integration_test_parse_logs.py
@@ -327,10 +327,13 @@ def main():
 
     if args.nearest:
         main_df = fuzzy_merge(lhs, rhs, keyA="name", keyB="name", threshold=80)
-        main_df["name_A"] = main_df["name"]
-        main_df["name_B"] = main_df["name"]
     else:
         main_df = lhs.merge(rhs, on="name", suffixes=("_A", "_B"), how="outer")
+        main_df["name_A"] = main_df["name"]
+        main_df["name_B"] = main_df["name"]
+
+    assert "name_A" in main_df.columns
+    assert "name_B" in main_df.columns
 
     for col in main_df:
         if col.startswith("img_path"):