import mesh
import sim
from energies.demag import Demag

def test_sim_basic_square():
    testmesh = mesh.Mesh((10, 20, 30), (3, 3, 3), scale=1e-9)
    a = sim.Sim(testmesh, 1e4)
    a.add_energy(Demag())

@@ -1,15 +1,22 @@
 import 
+joommf.
 mesh%0Aimp
@@ -19,16 +19,23 @@
 %0Aimport 
+joommf.
 sim%0Afrom
@@ -35,16 +35,23 @@
 im%0Afrom 
+joommf.
 energies

Fix test_sim.py