diff --git a/ci/cudf_pandas_scripts/third-party-integration/ci_run_library_tests.sh b/ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh similarity index 69% rename from ci/cudf_pandas_scripts/third-party-integration/ci_run_library_tests.sh rename to ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh index 54a56508cdc..d44d25d658c 100755 --- a/ci/cudf_pandas_scripts/third-party-integration/ci_run_library_tests.sh +++ b/ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh @@ -9,23 +9,17 @@ cleanup() { trap cleanup EXIT -runtest_gold() { +runtest() { local lib=$1 + local mode=$2 - pytest \ - -v \ - --continue-on-collection-errors \ - --cache-clear \ - --numprocesses=${NUM_PROCESSES} \ - --dist=worksteal \ - ${TEST_DIR}/test_${lib}*.py -} - -runtest_cudf_pandas() { - local lib=$1 + local plugin="" + if [ "$mode" = "cudf" ]; then + plugin="-p cudf.pandas" + fi pytest \ - -p cudf.pandas \ + $plugin \ -v \ --continue-on-collection-errors \ --cache-clear \ @@ -38,8 +32,8 @@ main() { local lib=$1 # generation phase - runtest_gold ${lib} - runtest_cudf_pandas ${lib} + runtest ${lib} "gold" + runtest ${lib} "cudf" # assertion phase pytest \ diff --git a/ci/cudf_pandas_scripts/third-party-integration/test.sh b/ci/cudf_pandas_scripts/third-party-integration/test.sh index 89b28c30e39..f8ddbaba0f3 100755 --- a/ci/cudf_pandas_scripts/third-party-integration/test.sh +++ b/ci/cudf_pandas_scripts/third-party-integration/test.sh @@ -72,7 +72,7 @@ main() { fi done - TEST_DIR=${TEST_DIR} NUM_PROCESSES=${NUM_PROCESSES} ci/cudf_pandas_scripts/third-party-integration/ci_run_library_tests.sh ${lib} + TEST_DIR=${TEST_DIR} NUM_PROCESSES=${NUM_PROCESSES} ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh ${lib} rapids-logger "Test script exiting with value: ${EXITCODE}" done