Use PYTEST_PYARGS to define module to test

apache · Jun 19, 2024 · 15ca182 · 15ca182
1 parent b0617b9
commit 15ca182
Show file tree

Hide file tree

Showing 3 changed files with 3 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

diff --git a/python/pyarrow/pandas_compat.py b/python/pyarrow/pandas_compat.py
@@ -1028,6 +1028,7 @@ def _is_generated_index_name(name):
     pattern = r'^__index_level_\d+__$'
     return re.match(pattern, name) is not None
 
+
 def get_pandas_logical_type_map():
     global _pandas_logical_type_map