Skip to content
Snippets Groups Projects
Commit 6a880afa authored by Jeff Zhang's avatar Jeff Zhang Committed by Josh Rosen
Browse files

[SPARK-12361][PYSPARK][TESTS] Should set PYSPARK_DRIVER_PYTHON before Python tests

Although this patch still doesn't solve the issue why the return code is 0 (see JIRA description), it resolves the issue of python version mismatch.

Author: Jeff Zhang <zjffdu@apache.org>

Closes #10322 from zjffdu/SPARK-12361.
parent d252b2d5
No related branches found
No related tags found
No related merge requests found
...@@ -56,7 +56,8 @@ LOGGER = logging.getLogger() ...@@ -56,7 +56,8 @@ LOGGER = logging.getLogger()
def run_individual_python_test(test_name, pyspark_python): def run_individual_python_test(test_name, pyspark_python):
env = dict(os.environ) env = dict(os.environ)
env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python)}) env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python),
'PYSPARK_DRIVER_PYTHON': which(pyspark_python)})
LOGGER.debug("Starting test(%s): %s", pyspark_python, test_name) LOGGER.debug("Starting test(%s): %s", pyspark_python, test_name)
start_time = time.time() start_time = time.time()
try: try:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment