[Test] Enable kvstore test (dmlc#3079)

* try enable kvstore test * fix * fix * seperate out kvstore test * add comment
anirband · Jul 2, 2021 · 0b3a621 · 0b3a621
1 parent 35ffceb
commit 0b3a621
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -57,7 +57,7 @@ def cpp_unit_test_win64() {
 def unit_test_linux(backend, dev) {
   init_git()
   unpack_lib("dgl-${dev}-linux", dgl_linux_libs)
-  timeout(time: 20, unit: 'MINUTES') {
+  timeout(time: 30, unit: 'MINUTES') {
     sh "bash tests/scripts/task_unit_test.sh ${backend} ${dev}"
   }
 }

diff --git a/tests/distributed/test_new_kvstore.py.bak → .../distributed/kv_store/test_new_kvstore.py b/tests/distributed/test_new_kvstore.py.bak → .../distributed/kv_store/test_new_kvstore.py
diff --git a/tests/distributed/test_dist_graph_store.py b/tests/distributed/test_dist_graph_store.py
@@ -176,6 +176,7 @@ def check_dist_emb(g, num_clients, num_nodes, num_edges):
     # Test sparse emb
     try:
         emb = NodeEmbedding(g.number_of_nodes(), 1, 'emb1', emb_init)
+        nids = F.arange(0, int(g.number_of_nodes()))
         lr = 0.001
         optimizer = SparseAdagrad([emb], lr=lr)
         with F.record_grad():

diff --git a/tests/scripts/task_unit_test.sh b/tests/scripts/task_unit_test.sh
@@ -37,5 +37,7 @@ python3 -m pytest -v --junitxml=pytest_backend.xml tests/$DGLBACKEND || fail "ba
 
 export OMP_NUM_THREADS=1
 if [ $2 != "gpu" ]; then
-    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed || fail "distributed"
+    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed/*.py || fail "distributed"
+    # Seperate kvstore test to another process, to avoid hangs
+    python3 -m pytest -v --capture=tee-sys --junitxml=pytest_distributed.xml tests/distributed/kv_store/*.py || fail "distributed kvstore"
 fi