support loading the test space in tck. (#5104)

vesoft-inc · Dec 23, 2022 · 53d26dc · 53d26dc
1 parent b5df05c
commit 53d26dc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -231,6 +231,9 @@ def load_ldbc_v0_3_3():
 def load_student_data():
     yield load_csv_data_once("student")
 
+@pytest.fixture(scope="session")
+def load_test_data():
+    yield load_csv_data_once("test")
 
 # TODO(yee): Delete this when we migrate all test cases
 @pytest.fixture(scope="class")

diff --git a/tests/tck/conftest.py b/tests/tck/conftest.py
@@ -182,6 +182,7 @@ def preload_space(
     load_nba_int_vid_data,
     load_student_data,
     load_ldbc_v0_3_3,
+    load_test_data,
     exec_ctx,
 ):
     space = normalize_outline_scenario(request, space)
@@ -193,6 +194,8 @@ def preload_space(
         exec_ctx["space_desc"] = load_student_data
     elif space == "ldbc_v0_3_3":
         exec_ctx["ldbc_v0_3_3"] = load_ldbc_v0_3_3
+    elif space == "test":
+        exec_ctx["test"] = load_test_data
     else:
         raise ValueError(f"Invalid space name given: {space}")