From 53d26dcfeffbefdbc6d9b51a96f044c5bf668ca1 Mon Sep 17 00:00:00 2001
From: Cheng Xuntao <7731943+xtcyclist@users.noreply.github.com>
Date: Fri, 23 Dec 2022 18:03:27 +0800
Subject: [PATCH] support loading the test space in tck. (#5104)

---
 tests/conftest.py     | 3 +++
 tests/tck/conftest.py | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/tests/conftest.py b/tests/conftest.py
index 6c8332bd81b..39ab47788d5 100644
--- 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
index 2b35433e921..294aa7f7d8a 100644
--- 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}")