Rework rel group (#5280)

acquamarin · ray6080 · andyfengHKU · web-flow · commit 31b0a757289b · 2025-05-27T18:10:35.000+08:00
Co-authored-by: Guodong Jin &lt;guod.jin@gmail.com&gt;
Co-authored-by: CI Bot &lt;ray6080@users.noreply.github.com&gt;
Co-authored-by: xiyang &lt;xiyangf1997@gmail.com&gt;
Co-authored-by: CI Bot &lt;andyfengHKU@users.noreply.github.com&gt;
diff --git a/test/test_datatype.py b/test/test_datatype.py
@@ -337,7 +337,7 @@ def test_recursive_rel(conn_db_readonly: ConnDB) -> None:
     assert len(e["_rels"]) == 1
     rel = e["_rels"][0]
     excepted_rel = {
-        "_id": {"offset": 0, "table": 4},
+        "_id": {"offset": 0, "table": 5},
         "_src": {"offset": 0, "table": 0},
         "_dst": {"offset": 0, "table": 1},
         "_label": "studyAt",
diff --git a/test/test_fsm.py b/test/test_fsm.py
@@ -97,8 +97,8 @@ def fsm_rel_group_setup(tmp_path: Path):
     conn.execute("create rel table likes (FROM personA TO personB, FROM personB To personA, date DATE);")
     conn.execute(f'COPY personA FROM "{KUZU_ROOT}/dataset/rel-group/node.csv";')
     conn.execute(f'COPY personB FROM "{KUZU_ROOT}/dataset/rel-group/node.csv";')
-    conn.execute(f'COPY likes_personA_personB FROM "{KUZU_ROOT}/dataset/rel-group/edge.csv";')
-    conn.execute(f'COPY likes_personB_personA FROM "{KUZU_ROOT}/dataset/rel-group/edge.csv";')
+    conn.execute(f'COPY likes FROM "{KUZU_ROOT}/dataset/rel-group/edge.csv" (FROM="personA", TO="personB");')
+    conn.execute(f'COPY likes FROM "{KUZU_ROOT}/dataset/rel-group/edge.csv" (FROM="personB", TO="personA");')
     return db, conn
 
 
@@ -186,9 +186,7 @@ def test_fsm_reclaim_struct(fsm_rel_table_setup) -> None:
 
 def test_fsm_reclaim_rel_group(fsm_rel_group_setup) -> None:
     _, conn = fsm_rel_group_setup
-    used_pages = get_used_page_ranges(conn, "likes_personA_personB") + get_used_page_ranges(
-        conn, "likes_personB_personA"
-    )
+    used_pages = get_used_page_ranges(conn, "likes")
     conn.execute("drop table likes")
     prevent_data_file_truncation(conn)
     conn.execute("checkpoint")
@@ -199,10 +197,8 @@ def test_fsm_reclaim_rel_group(fsm_rel_group_setup) -> None:
 def test_fsm_reclaim_rel_group_column(fsm_rel_group_setup) -> None:
     _, conn = fsm_rel_group_setup
     used_pages = (
-        get_used_page_ranges(conn, "likes_personA_personB", "fwd_date")
-        + get_used_page_ranges(conn, "likes_personA_personB", "bwd_date")
-        + get_used_page_ranges(conn, "likes_personB_personA", "fwd_date")
-        + get_used_page_ranges(conn, "likes_personB_personA", "bwd_date")
+        get_used_page_ranges(conn, "likes", "fwd_date")
+        + get_used_page_ranges(conn, "likes", "bwd_date")
     )
     conn.execute("alter table likes drop date")
     prevent_data_file_truncation(conn)
