def test_exch_energy_1d(): mesh = CuboidMesh(nx=2, ny=1, nz=1) sim = Sim(mesh) exch = UniformExchange(1.23) sim.add(exch) sim.set_m((0, 0, 1)) energy = exch.compute_energy() assert energy == -1.23