Merge pull request #129 from keisukefukuda/bugfix

bugfix on exec
keisukefukuda · May 7, 2019 · 8bfbba8 · 8bfbba8
2 parents f849cfe + 0836e52
commit 8bfbba8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/mpienv/mpibase.py b/mpienv/mpibase.py
@@ -112,6 +112,7 @@ def split_mpi_user_prog(cmds):
         '--preload-files-dest-dir',
         '--tmpdir', '-wd', '-wdir', '-x',
         '-tune', '--tune',
+        '--timeout',
         '-aborted', '--aborted',
         '--app', '-cf', '--cartofile',
         '-ompi-server', '--ompi-server',

diff --git a/tests/test_main.sh b/tests/test_main.sh
@@ -405,7 +405,7 @@ EOF
 }
 
 
-test_reg_issue10(){
+test_reg_issue10() {
     # Regression test for #10
     # https://github.com/keisukefukuda/mpienv/issues/10
     assertSuccess mpienv autodiscover -q --add ${MPI_PREFIX}
@@ -420,6 +420,17 @@ test_reg_issue10(){
     assertEquals "\$OUT must be empty" "" "${OUT}"
 }
 
+test_reg_issue_timeout() {
+    # Regression test for
+    # https://github.com/keisukefukuda/mpienv/issues/10
+    assertSuccess mpienv autodiscover -q --add ${MPI_PREFIX}
+
+    mpienv use ${OMPI} # this command should install mpi4py to mpich-3.2
+
+    mpienv exec -host localhost:2 --timeout 200 -n 2 $PYTHON -c "from mpi4py import MPI"
+    assertTrue "$LINENO: import mpi4py should success with --timeout option" $?
+}
+
 # suite() {
 #     suite_addTest "test_mpi4py"
 #     # suite_addTest "test_mpicc"