Use pytest-xdist worksteal in wheel tests.

rapidsai · Mar 5, 2024 · 9b86518 · 9b86518
1 parent 3ec831e
commit 9b86518
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/ci/test_wheel_cuproj.sh b/ci/test_wheel_cuproj.sh
@@ -22,5 +22,13 @@ python -m pip install $(echo ./dist/cuproj*.whl)[test]
 if [[ "$(arch)" == "aarch64" && ${RAPIDS_BUILD_TYPE} == "pull-request" ]]; then
     python ./ci/wheel_smoke_test_cuproj.py
 else
-    python -m pytest -n 8 ./python/cuproj/cuproj/tests
+    rapids-logger "pytest cuproj"
+    pushd python/cuproj/cuproj
+    python -m pytest \
+      --cache-clear \
+      --junitxml="${RAPIDS_TESTS_DIR}/junit-cuproj.xml" \
+      --numprocesses=8 \
+      --dist=worksteal \
+      tests
+    popd
 fi
diff --git a/ci/test_wheel_cuspatial.sh b/ci/test_wheel_cuspatial.sh
@@ -18,5 +18,13 @@ python -m pip install $(echo ./dist/cuspatial*.whl)[test]
 if [[ "$(arch)" == "aarch64" && ${RAPIDS_BUILD_TYPE} == "pull-request" ]]; then
     python ./ci/wheel_smoke_test_cuspatial.py
 else
-    python -m pytest -n 8 ./python/cuspatial/cuspatial/tests
+    rapids-logger "pytest cuspatial"
+    pushd python/cuspatial/cuspatial
+    python -m pytest \
+      --cache-clear \
+      --junitxml="${RAPIDS_TESTS_DIR}/junit-cuproj.xml" \
+      --numprocesses=8 \
+      --dist=worksteal \
+      tests
+    popd
 fi