diff --git a/test/test_meshmode.py b/test/test_meshmode.py index f25930ca24fafc6660622d459aa4716ed088f47e..be66f28dab3ea82666e74c7475c4d675cb5c939f 100644 --- a/test/test_meshmode.py +++ b/test/test_meshmode.py @@ -52,12 +52,14 @@ logger = logging.getLogger(__name__) @pytest.mark.parametrize("num_partitions", [3, 6]) def test_mpi_communication(num_partitions): num_ranks = num_partitions + 1 - import subprocess, sys, os + from subprocess import check_call + import sys + import os newenv = os.environ.copy() newenv["PYTOOLS_RUN_WITHIN_MPI"] = "1" - subprocess.check_call(["mpirun", "-np", str(num_ranks), - sys.executable, "testmpi.py", str(num_partitions)], - env=newenv) + check_call(["mpirun", "-np", str(num_ranks), + sys.executable, "test/testmpi.py", str(num_partitions)], + env=newenv) # {{{ partition_interpolation diff --git a/testmpi.py b/test/testmpi.py similarity index 98% rename from testmpi.py rename to test/testmpi.py index 87a51914c0a729235823717a4eab29edc1c4106c..7f84e5015a61743afbd9e63b08a03d314dc5a7b2 100644 --- a/testmpi.py +++ b/test/testmpi.py @@ -1,3 +1,4 @@ +from __future__ import division, absolute_import, print_function import numpy as np