allow for non-mpi tests

This commit is contained in:
Peter Zhokhov
2019-01-24 14:35:41 -08:00
parent 547764efc9
commit e868bdaa1a

View File

@@ -5,7 +5,10 @@ import cloudpickle
import base64 import base64
import pytest import pytest
from mpi4py import MPI try:
from mpi4py import MPI
except ImportError:
MPI = None
def test_with_mpi(nproc=2, timeout=30, skip_if_no_mpi=True): def test_with_mpi(nproc=2, timeout=30, skip_if_no_mpi=True):
def outer_thunk(fn): def outer_thunk(fn):