mpi_sens_test.py 文件源码

def test_multi_solve(models, sim):

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

    results = sim.multi_solve(models, ['simp',], dofs)

    for i, res in enumerate(results):
        npt.assert_array_equal(
            res[1][0],
            np.arange(10) * (np.arange(10) + i + 1)
        )