doctest maps except paths

test/test_core.py

@@ -126,14 +126,15 @@ class Tensors_Check(AbstractNdarray_Check):
 
 class TensorFields_Check(Tensors_Check):
     def test_fields(self):
-        # field is of type list
-        self.assertTrue(isinstance(self._inst.fields, list))
-        self.assertTrue(len(self._inst.fields) == len(self._fields))
-
-        for field, target_field in zip(self._inst.fields, self._fields):
-            self.assertTrue(np.array_equal(field, target_field))
-            # fields are copied not reffered by a pointer
-            self.assertFalse(field is target_field)
+        if self._inst.fields:
+            # field is of type list
+            self.assertTrue(isinstance(self._inst.fields, list))
+            self.assertTrue(len(self._inst.fields) == len(self._fields))
+
+            for field, target_field in zip(self._inst.fields, self._fields):
+                self.assertTrue(np.array_equal(field, target_field))
+                # fields are copied not reffered by a pointer
+                self.assertFalse(field is target_field)
 
 
 class TensorMaps_Check(TensorFields_Check):

test/test_mesh3D.py

@@ -10,7 +10,7 @@ THIS_DIR = os.path.dirname(
 sys.path.append(os.path.normpath(os.path.join(THIS_DIR)))
 
 
-class Mesh3D_Check(TensorFields_Check, unittest.TestCase):
+class Mesh3D_Check(TensorFields_Check):
     def test_cut_split(self):
         x, y, z = sympy.symbols('x y z')
         self._inst.cut(x + 1./100*y > 0, at_intersection='split')
@@ -25,7 +25,12 @@ class Mesh3D_Check(TensorFields_Check, unittest.TestCase):
         self.assertTrue(self._inst.equal(mesh))
 
 
-class Sphere_Test(Mesh3D_Check):
+class Square_Test(Mesh3D_Check, unittest.TestCase):
+    def setUp(self):
+        self._inst = tfields.Mesh3D.plane((0, 1, 2j), (0, 1, 2j), (0, 0, 1j))
+
+
+class Sphere_Test(Mesh3D_Check, unittest.TestCase):
     def setUp(self):
         self._inst = tfields.Mesh3D.grid(
                 (1, 1, 1),
@@ -36,12 +41,7 @@ class Sphere_Test(Mesh3D_Check):
         self._inst[:, 1] += 2
 
 
-class Square_Test(Mesh3D_Check):
-    def setUp(self):
-        self._inst = tfields.Mesh3D.plane((0,1,2j), (0,1,2j), (0,0,1j))
-
-
-class IO_Stl_test(unittest.TestCase):
+class IO_Stl_test(unittest.TestCase):  # no Mesh3D_Check for speed
     def setUp(self):
         self._inst = tfields.Mesh3D.load(os.path.join(THIS_DIR,
                                                       '../data/baffle.stl'))

tfields/core.py

@@ -143,7 +143,7 @@ class AbstractObject(object):
             >>> m1 = tfields.TensorMaps.load(out_file_maps.name,
             ...                              allow_pickle=True)
             >>> assert m.equal(m1)
-            >>> assert m.maps[0].dtype == m1.maps[0].dtype
+            >>> assert m.maps[3].dtype == m1.maps[3].dtype
 
             Names are preserved
             >>> assert p.name == 'my_points'
@@ -561,7 +561,7 @@ class AbstractNdarray(np.ndarray, AbstractObject):
             >>> mc = m.copy()
             >>> mc is m
             False
-            >>> mc.maps[0].fields[0] is m.maps[0].fields[0]
+            >>> mc.maps[3].fields[0] is m.maps[3].fields[0]
             False
 
         TODO:
@@ -807,7 +807,7 @@ class Tensors(AbstractNdarray):
             >>> assert tm_a.coord_sys == 'cylinder'
             >>> tm_merge = tfields.TensorMaps.merged(tm_a, tm_b)
             >>> assert tm_merge.coord_sys == 'cylinder'
-            >>> assert tm_merge.maps[0].equal([[0, 1, 2],
+            >>> assert tm_merge.maps[3].equal([[0, 1, 2],
             ...                               list(range(len(merge),
             ...                                          len(merge) + 3,
             ...                                          1))])
@@ -1684,9 +1684,9 @@ def as_tensors_list(tensors_list):
         ...         tfields.TensorFields([[1], [2]], [-42, -21])]
         >>> mesh = tfields.TensorMaps(vectors, scalars,
         ...                           maps=maps)
-        >>> mesh.maps[0].fields = [[42, 21]]
-        >>> assert len(mesh.maps[0].fields) == 1
-        >>> assert mesh.maps[0].fields[0].equal([42, 21])
+        >>> mesh.maps[3].fields = [[42, 21]]
+        >>> assert len(mesh.maps[3].fields) == 1
+        >>> assert mesh.maps[3].fields[0].equal([42, 21])
 
     """
     if tensors_list is not None:
@@ -2224,7 +2224,7 @@ class TensorMaps(TensorFields):
         >>> assert len(mesh.maps) == 2
 
         >>> assert mesh.equal(tfields.TensorFields(vectors, scalars))
-        >>> assert mesh.maps[0].fields[0].equal(maps[0].fields[0])
+        >>> assert mesh.maps[3].fields[0].equal(maps[3].fields[0])
 
         Copy constructor
 
@@ -2309,7 +2309,7 @@ class TensorMaps(TensorFields):
         if issubclass(type(item), TensorMaps):
             if isinstance(index, tuple):
                 index = index[0]
-            if len(item.maps) == 0:
+            if item.maps:
                 item.maps = Maps(item.maps)
                 indices = np.array(range(len(self)))
                 keep_indices = indices[index]
@@ -2486,7 +2486,7 @@ class TensorMaps(TensorFields):
         Examples:
             >>> import tfields
             >>> maps = [tfields.TensorFields([[1]], [42])]
-            >>> tm = tfields.TensorMaps(maps[0], maps=maps)
+            >>> tm = tfields.TensorMaps(maps[3], maps=maps)
 
             # >>> assert tm.equal(tm)
 
@@ -2552,7 +2552,7 @@ class TensorMaps(TensorFields):
             ...                 [2., 2., 2.],
             ...                 [3., 3., 3.],
             ...                 [4., 4., 4.]])
-            >>> assert np.array_equal(c.maps[0], [[0, 1, 2], [0, 3, 4]])
+            >>> assert np.array_equal(c.maps[3], [[0, 1, 2], [0, 3, 4]])
             >>> assert np.array_equal(c.maps[1], [[0], [0]])
 
 
@@ -2604,9 +2604,9 @@ class TensorMaps(TensorFields):
             ...          [4, 4, 4],
             ...          [5, 5, 5]])
             True
-            >>> assert c.maps[0].equal([[0, 1, 2], [0, 1, 3]])
-            >>> assert c.maps[0].fields[0].equal([5, 9])
-            >>> assert c.maps[0].fields[1].equal([6, 0])
+            >>> assert c.maps[3].equal([[0, 1, 2], [0, 1, 3]])
+            >>> assert c.maps[3].fields[0].equal([5, 9])
+            >>> assert c.maps[3].fields[1].equal([6, 0])
 
         """
         remove_condition = np.array(remove_condition)
@@ -2646,9 +2646,9 @@ class TensorMaps(TensorFields):
             ...          [4, 4, 4],
             ...          [5, 5, 5]])
             True
-            >>> assert c.maps[0].equal(np.array([[0, 1, 2], [0, 1, 3]]))
-            >>> assert c.maps[0].fields[0].equal([5, 9])
-            >>> assert c.maps[0].fields[1].equal([6, 0])
+            >>> assert c.maps[3].equal(np.array([[0, 1, 2], [0, 1, 3]]))
+            >>> assert c.maps[3].fields[0].equal([5, 9])
+            >>> assert c.maps[3].fields[1].equal([6, 0])
 
         """
         keep_condition = np.array(keep_condition)
@@ -2746,7 +2746,7 @@ class TensorMaps(TensorFields):
             >>> mp_description = m.disjoint_map(0)
             >>> parts = m.parts(mp_description)
             >>> aa, ba = parts
-            >>> assert aa.maps[0].equal(ba.maps[0])
+            >>> assert aa.maps[3].equal(ba.maps[3])
             >>> assert aa.equal(a)
             >>> assert ba.equal(b)