From 76bdc3fe6d7c7316f5fad9ca964ccfc8bc1b9c18 Mon Sep 17 00:00:00 2001 From: "Harris.Chu" <1726587+HarrisChu@users.noreply.github.com> Date: Wed, 13 Oct 2021 10:13:47 +0800 Subject: [PATCH] enhance tck data (#3042) Co-authored-by: Yichen Wang <18348405+Aiee@users.noreply.github.com> --- tests/nebula-test-run.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/nebula-test-run.py b/tests/nebula-test-run.py index 2f1fbc371fe..c012cbca2fe 100755 --- a/tests/nebula-test-run.py +++ b/tests/nebula-test-run.py @@ -108,8 +108,11 @@ def start_nebula(nb, configs): with open(SPACE_TMP_PATH, "w") as f: spaces = [] - for space in ("nba", "nba_int_vid", "student"): - data_dir = os.path.join(CURR_PATH, "data", space) + folder = os.path.join(CURR_PATH, "data") + for space in os.listdir(folder): + if not os.path.exists(os.path.join(folder, space, "config.yaml")): + continue + data_dir = os.path.join(folder, space) space_desc = load_csv_data(sess, data_dir, space) spaces.append(space_desc.__dict__) f.write(json.dumps(spaces))