Commit 309bb161 authored by dboe's avatar dboe
Browse files

doctest maps except paths

parent ab1b51d9
......@@ -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):
......
......@@ -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'))
......
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment