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)