|
@@ -510,3 +510,17 @@ def test_remove_unused_vertices_single():
|
|
|
assert all(vertex_index < len(surface.vertices)
|
|
|
for triangle in surface.triangles
|
|
|
for vertex_index in triangle.vertex_indices)
|
|
|
+
|
|
|
+
|
|
|
+def test_select_vertices():
|
|
|
+ surface = Surface()
|
|
|
+ for i in range(4):
|
|
|
+ surface.add_vertex(Vertex(i, i, i))
|
|
|
+ assert (surface.select_vertices([2, 1])
|
|
|
+ == [surface.vertices[2], surface.vertices[1]]).all()
|
|
|
+ assert (surface.select_vertices((3, 2))
|
|
|
+ == [surface.vertices[3], surface.vertices[2]]).all()
|
|
|
+ assert (surface.select_vertices((3, 2))
|
|
|
+ == [[3, 3, 3], [2, 2, 2]]).all()
|
|
|
+ assert (surface.select_vertices(filter(lambda i: i % 2 == 1, range(4)))
|
|
|
+ == [[1, 1, 1], [3, 3, 3]]).all()
|