From d723aa79dd1d6bfd2d5eacb2bd1980f4ce63e3b6 Mon Sep 17 00:00:00 2001
From: HarrisChu <1726587+HarrisChu@users.noreply.github.com>
Date: Tue, 12 Oct 2021 15:56:49 +0800
Subject: [PATCH] enhance tck data

---
 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))