tests

Author: tomvanmele

src/compas/datastructures/halfedge/halfedge.py

@@ -260,50 +260,50 @@ def get_any_face_vertex(self, fkey):
         """
         return self.face_vertices(fkey)[0]
 
-    def vertex_sample(self, n=1):
+    def vertex_sample(self, size=1):
         """A random sample of the vertices.
 
         Parameters
         ----------
-        n : int, optional
+        size : int, optional
             The number of vertices in the random sample.
 
         Returns
         -------
         list
             The identifiers of the vertices.
         """
-        return sample(list(self.vertices()), n)
+        return sample(list(self.vertices()), size)
 
-    def edge_sample(self, n=1):
+    def edge_sample(self, size=1):
         """A random sample of the edges.
 
         Parameters
         ----------
-        n : int, optional
+        size : int, optional
             The number of edges in the random sample.
 
         Returns
         -------
         list
             The identifiers of the edges.
         """
-        return sample(list(self.edges()), n)
+        return sample(list(self.edges()), size)
 
-    def face_sample(self, n=1):
+    def face_sample(self, size=1):
         """A random sample of the faces.
 
         Parameters
         ----------
-        n : int, optional
+        size : int, optional
             The number of faces in the random sample.
 
         Returns
         -------
         list
             The identifiers of the faces.
         """
-        return sample(list(self.faces()), n)
+        return sample(list(self.faces()), size)
 
     def key_index(self):
         """Returns a dictionary that maps vertex dictionary keys to the

tests/compas/datastructures/test_graph.py

@@ -52,6 +52,25 @@ def test_graph_json_schema(graph):
     graph.validate_json()
 
 
+# ==============================================================================
+# Tests - Samples
+# ==============================================================================
+
+
+def test_node_sample(graph):
+    for node in graph.node_sample():
+        assert graph.has_node(node)
+    for node in graph.node_sample(size=graph.number_of_nodes()):
+        assert graph.has_node(node)
+
+
+def test_edge_sample(graph):
+    for edge in graph.edge_sample():
+        assert graph.has_edge(*edge)
+    for edge in graph.edge_sample(size=graph.number_of_edges()):
+        assert graph.has_edge(*edge)
+
+
 # ==============================================================================
 # Tests - Attributes
 # ==============================================================================

tests/compas/datastructures/test_halfedge.py

@@ -65,6 +65,32 @@ def test_json_schema(mesh):
         mesh.validate_json()
 
 
+# ==============================================================================
+# Tests - Samples
+# ==============================================================================
+
+
+def test_vertex_sample(mesh):
+    for vertex in mesh.vertex_sample():
+        assert mesh.has_vertex(vertex)
+    for vertex in mesh.vertex_sample(size=mesh.number_of_vertices()):
+        assert mesh.has_vertex(vertex)
+
+
+def test_edge_sample(mesh):
+    for edge in mesh.edge_sample():
+        assert mesh.has_edge(edge)
+    for edge in mesh.edge_sample(size=mesh.number_of_edges()):
+        assert mesh.has_edge(edge)
+
+
+def test_face_sample(mesh):
+    for face in mesh.face_sample():
+        assert mesh.has_face(face)
+    for face in mesh.face_sample(size=mesh.number_of_faces()):
+        assert mesh.has_face(face)
+
+
 # ==============================================================================
 # Tests - Vertex Attributes
 # ==============================================================================