all mpi4py tests in a single py file

.travis.yml

@@ -20,5 +20,5 @@ before_install:
 
 script:
   - docker-compose scale mpi_head=1 mpi_node=${NNODES} # &> logfile &
-  - docker-compose exec --privileged mpi_head /bin/bash -c ./home/mpirun/mpi4py_benchmarks/run_tests.sh ${NNODES}
+  - docker-compose exec --privileged mpi_head mpirun -n ${NNODES} python /home/mpirun/mpi4py_benchmarks/all_tests.py
   - docker-compose down

Dockerfile

@@ -53,7 +53,7 @@ RUN chmod -R 600 ${SSHDIR}* && \
     chown -R ${USER}:${USER} ${SSHDIR}
 
 # ------------------------------------------------------------
-# Copy Rosa's MPI4PY example scripts
+# Copy MPI4PY example scripts
 # ------------------------------------------------------------
 
 ADD mpi4py_benchmarks ${HOME}/mpi4py_benchmarks

makefile

@@ -21,5 +21,5 @@ rebuild:
 
 main:
 	docker-compose scale mpi_head=1 mpi_node=${NNODES}
-	docker-compose exec --privileged mpi_head /bin/bash -c ./home/mpirun/mpi4py_benchmarks/run_tests.sh ${NNODES}
+	docker-compose exec --privileged mpi_head mpirun -n ${NNODES} python /home/mpirun/mpi4py_benchmarks/all_tests.py
 	docker-compose down

mpi4py_benchmarks/matrix_vector_product.py

@@ -18,7 +18,7 @@
 from __future__ import division
 
 import numpy as np
-from numpy.fft import fft2, ifft2
+# from numpy.fft import fft2, ifft2
 from math import ceil, fabs
 from mpi4py import MPI
 
@@ -34,7 +34,7 @@ def pprint(str="", end="\n", comm=MPI.COMM_WORLD):
 # Main
 
 size = 10000           # length of vector v
-iter = 20              # number of iterations to run
+iter = 2000            # number of iterations to run
 
 counter = 0