mpi_sens_test.py 文件源码

def test_multi_solve_pll(models, sim):

    dofs = []
    for i in range(20):
        dofs.append(models['simp'].get_dof() + i)
    # end

    results = sim.multi_solve_mpi(models, ['simp',], dofs, verb=False)


    # for i in results:
    #     print('rank {:d} item{:d} {:}'.format(myrank, i, results[i][1][0]))

    for i, res in enumerate(results):        
        npt.assert_array_equal(
            res[1][0],
            np.arange(10) * (np.arange(10) + i + 1),
            err_msg='Error in {:d} dof set'.format(i)
        )