Skip to content

Commit

Permalink
support loading the test space in tck. (vesoft-inc#1980)
Browse files Browse the repository at this point in the history
Co-authored-by: Cheng Xuntao <7731943+xtcyclist@users.noreply.github.com>
  • Loading branch information
nebula-bots and xtcyclist authored Dec 26, 2022
1 parent 02d5f9e commit 915d3b7
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down
3 changes: 3 additions & 0 deletions tests/tck/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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}")

Expand Down

0 comments on commit 915d3b7

Please sign in to comment.