def test_m_average(): mesh = CuboidMesh(nx=3, ny=4, nz=5) sim = Sim(mesh) sim.set_m((0, 0, 1)) a = sim.compute_average() assert a[2] == 1.0