old_contents: import unittest
  import numpy as np
  from finmag.util.oommf import mesh
  
  
  class TestIterCoordsInt(unittest.TestCase):
  
      def test_zyx_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1))
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [2, 0, 0]]
          assert np.array_equal(m.mesh_size, [3, 1, 1])
          assert m.array_order == mesh.Mesh.ZYX
          assert np.array_equal(expected, indices)
  
          m = mesh.Mesh((2, 2, 2), cellsize=(1, 1, 1))
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0],
                      [0, 0, 1], [1, 0, 1], [0, 1, 1], [1, 1, 1]]
          assert np.array_equal(m.mesh_size, [2, 2, 2])
          assert m.array_order == mesh.Mesh.ZYX
          assert np.array_equal(expected, indices)
  
      def test_xyz_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1), array_order=mesh.Mesh.XYZ)
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [2, 0, 0]]
          assert np.array_equal(m.mesh_size, [3, 1, 1])
          assert m.array_order == mesh.Mesh.XYZ
          assert np.array_equal(expected, indices)
  
          m = mesh.Mesh((2, 2, 2), cellsize=(1, 1, 1), array_order=mesh.Mesh.XYZ)
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1],
                      [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]
          assert np.array_equal(m.mesh_size, [2, 2, 2])
          assert m.array_order == mesh.Mesh.XYZ
          assert np.array_equal(expected, indices)
  
  
  class TestIterCoords(unittest.TestCase):
  
      def test_zyx_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1))
          coords = [r for r in m.iter_coords()]
          expected = [[0.5, 0.5, 0.5], [1.5, 0.5, 0.5], [2.5, 0.5, 0.5]]
          assert np.array_equal(expected, coords)
  
new_contents: import unittest
  import numpy as np
  import mesh
  
  
  class TestIterCoordsInt(unittest.TestCase):
  
      def test_zyx_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1))
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [2, 0, 0]]
          assert np.array_equal(m.mesh_size, [3, 1, 1])
          assert m.array_order == mesh.Mesh.ZYX
          assert np.array_equal(expected, indices)
  
          m = mesh.Mesh((2, 2, 2), cellsize=(1, 1, 1))
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0],
                      [0, 0, 1], [1, 0, 1], [0, 1, 1], [1, 1, 1]]
          assert np.array_equal(m.mesh_size, [2, 2, 2])
          assert m.array_order == mesh.Mesh.ZYX
          assert np.array_equal(expected, indices)
  
      def test_xyz_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1), array_order=mesh.Mesh.XYZ)
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [1, 0, 0], [2, 0, 0]]
          assert np.array_equal(m.mesh_size, [3, 1, 1])
          assert m.array_order == mesh.Mesh.XYZ
          assert np.array_equal(expected, indices)
  
          m = mesh.Mesh((2, 2, 2), cellsize=(1, 1, 1), array_order=mesh.Mesh.XYZ)
          indices = [r for r in m.iter_coords_int()]
          expected = [[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1],
                      [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]
          assert np.array_equal(m.mesh_size, [2, 2, 2])
          assert m.array_order == mesh.Mesh.XYZ
          assert np.array_equal(expected, indices)
  
  
  class TestIterCoords(unittest.TestCase):
  
      def test_zyx_ordering(self):
          m = mesh.Mesh((3, 1, 1), cellsize=(1, 1, 1))
          coords = [r for r in m.iter_coords()]
          expected = [[0.5, 0.5, 0.5], [1.5, 0.5, 0.5], [2.5, 0.5, 0.5]]
          assert np.array_equal(expected, coords)
  
subject: fix import command