Use PYTEST_ARGS to define module to test

apache · Jun 19, 2024 · 2d9bf92 · 2d9bf92
1 parent a264453
commit 2d9bf92
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/ci/scripts/python_test.sh b/ci/scripts/python_test.sh
@@ -69,8 +69,4 @@ export PYARROW_TEST_PARQUET_ENCRYPTION
 export PYARROW_TEST_S3
 
 # Testing PyArrow
-if [ -z "${TEST_COMMAND}" ]; then
-  pytest -r s ${PYTEST_ARGS} --pyargs pyarrow
-else
-  ${TEST_COMMAND}
-fi
+pytest -r s ${PYTEST_ARGS} --pyargs ${PYTEST_PYARGS:-'pyarrow'}
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1247,7 +1247,7 @@ services:
       PARQUET_REQUIRE_ENCRYPTION:  # inherit
       # In the future we are supposed to run all tests.
       # This is only temporary.
-      TEST_COMMAND:  "pytest -r s --pyargs pyarrow.tests.test_array"
+      PYTEST_PYARGS:  "pyarrow.tests.test_array"
       HYPOTHESIS_PROFILE:  # inherit
       PYARROW_TEST_HYPOTHESIS:  # inherit
     volumes: *conda-volumes