chore: add test_wheel script

rapidsai · Dec 13, 2024 · 76f1886 · 76f1886
1 parent f286c95
commit 76f1886
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -0,0 +1,12 @@
+#!/bin/bash
+# Copyright (c) 2023, NVIDIA CORPORATION.
+
+set -eou pipefail
+
+RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
+RAPIDS_PY_WHEEL_NAME="dask_cuda-${RAPIDS_PY_CUDA_SUFFIX}" RAPIDS_PY_WHEEL_PURE="1" rapids-download-wheels-from-s3 ./dist
+
+# echo to expand wildcard before adding `[extra]` requires for pip
+python -m pip install $(echo ./dist/dask_cuda*.whl)[test]
+
+python -m pytest -n 8 ./python/dask_cuda/tests