def test_args():
# direct args
mock_mpiexec(2, f, "abc")
mock_mpiexec(2, f, "abc", index="cat", index2="cat")
# if have args as list and kwargs as dict, can dereference them.
args = ["abc"]
mock_mpiexec(2, f, *args)
kwargs = {"index": "cat", "index2": "cat"}
mock_mpiexec(2, f, *args, **kwargs)
# can also dereference as literals, which is parobably weird, but allowed.
mock_mpiexec(2, f, *["abc"], **{"index": "cat", "index2": "cat"})
def test_dask():
core_dask(None)
mockmpi.mock_mpiexec(3, core_dask)
mockmpi.mock_mpiexec(5, core_dask)
def test_mean_weights(nbin, ndata, nproc, mode, sparse):
mock_mpiexec(nproc, run_mean_weights, nbin, ndata, mode, sparse)
def test_mean(nbin, ndata, nproc, mode):
mock_mpiexec(nproc, run_mean, nbin, ndata, mode)
def test_mpi_session():
mock_mpiexec(2, run_mpi_session)
mock_mpiexec(4, run_mpi_session)
def test_mean_variance(nbin, ndata, nproc, mode):
mock_mpiexec(nproc, run_mean_variance, nbin, ndata, mode)
def test_mpi_kk():
output = __name__ == '__main__'
mock_mpiexec(4, do_mpi_kk, output)
mock_mpiexec(1, do_mpi_kk, output)
def test_sum(nbin, ndata, nproc, mode):
mock_mpiexec(nproc, run_sum, nbin, ndata, mode)
