Add a few unit tests

Author: gonzalocasas

src/compas/datastructures/assembly/assembly.py

@@ -130,9 +130,21 @@ def parts(self):
         for node in self.graph.nodes():
             yield self.graph.node_attribute(node, 'part')
 
-    def connections(self):
-        """The connections between the parts."""
-        return self.graph.edges()
+    def connections(self, data=False):
+        """Iterate over the connections between the parts.
+
+        Parameters
+        ----------
+        data : bool, optional
+            If ``True``, yield both the identifier and the attributes of each connection.
+
+        Yields
+        ------
+        tuple
+            The next connection identifier (u, v), if ``data`` is ``False``.
+            Otherwise, the next connector identifier and its attributes as a ((u, v), attr) tuple.
+        """
+        return self.graph.edges(data)
 
     def find(self, guid):
         """Find a part in the assembly by its GUID.

src/compas/datastructures/assembly/part.py

@@ -134,7 +134,8 @@ def frame(self, frame):
 
     @property
     def geometry(self):
-        # this is a temp solution
+        # TODO: this is a temp solution
+        # TODO: add memoization or some other kind of caching
         A = Mesh.from_shape(self.shape)
         for shape, operation in self.features:
             A.quads_to_triangles()

tests/compas/datastructures/test_assembly.py

@@ -0,0 +1,43 @@
+from compas.datastructures import Assembly, assembly
+from compas.datastructures import Part
+
+
+def test_init():
+    assembly = Assembly(name='abc')
+    assert assembly.name == 'abc'
+
+    assembly = Assembly(attr1='value', attr2=3.14)
+    assert assembly.attributes['attr1'] == 'value'
+    assert assembly.attributes['attr2'] == 3.14
+
+
+def test_add_parts():
+    assembly = Assembly()
+
+    for _ in range(3):
+        assembly.add_part(Part())
+
+    assert len(list(assembly.parts())) == 3
+
+
+def test_add_connections():
+    assembly = Assembly()
+    parts = [Part() for i in range(3)]
+
+    for part in parts:
+        assembly.add_part(part)
+
+    assembly.add_connection(parts[0], parts[1])
+    assembly.add_connection(parts[1], parts[2])
+    assembly.add_connection(parts[2], parts[0])
+
+    assert list(assembly.connections()) == [(0, 1), (1, 2), (2, 0)]
+
+
+def test_find():
+    assembly = Assembly()
+    part = Part()
+    assert assembly.find(part.guid) is None
+
+    assembly.add_part(part)
+    assert assembly.find(part.guid) == part