tests for str and repr

Author: yck011522

src/compas/geometry/polygon.py

@@ -105,7 +105,7 @@ def __repr__(self):
         return "{0}(points={1!r})".format(type(self).__name__, self.points)
 
     def __str__(self):
-        return "{0}(points={1})".format(type(self).__name__, [str(point) for point in self.points])
+        return "{0}(points=[{1}])".format(type(self).__name__, ", ".join([str(point) for point in self.points]))
 
     def __len__(self):
         return len(self.points)

tests/compas/geometry/test_frame.py

@@ -55,6 +55,12 @@ def test_frame_data():
         assert Frame.validate_data(other.__data__)
 
 
+def test_frame_str_repr():
+    frame = Frame.worldXY()
+    assert str(frame) == "Frame(point=Point(x=0.000, y=0.000, z=0.000), xaxis=Vector(x=1.000, y=0.000, z=0.000), yaxis=Vector(x=0.000, y=1.000, z=0.000))"
+    assert repr(frame) == "Frame(point=Point(x=0.0, y=0.0, z=0.0), xaxis=Vector(x=1.0, y=0.0, z=0.0), yaxis=Vector(x=0.0, y=1.0, z=0.0))"
+
+
 def test_frame_predefined():
     frame = Frame.worldXY()
     assert frame.point == Point(0, 0, 0)

tests/compas/geometry/test_plane.py

@@ -46,6 +46,12 @@ def test_plane_data():
         assert Plane.validate_data(other.__data__)
 
 
+def test_plane_str_repr():
+    plane = Plane.worldXY()
+    assert str(plane) == "Plane(point=Point(x=0.000, y=0.000, z=0.000), normal=Vector(x=0.000, y=0.000, z=1.000))"
+    assert repr(plane) == "Plane(point=Point(x=0.0, y=0.0, z=0.0), normal=Vector(x=0.0, y=0.0, z=1.0))"
+
+
 def test_plane_predefined():
     plane = Plane.worldXY()
     assert plane.point == Point(0, 0, 0)

tests/compas/geometry/test_pointcloud.py

@@ -39,6 +39,13 @@ def test_pointcloud_data():
         assert Pointcloud.validate_data(other.__data__)
 
 
+def test_pointcloud_str_repr():
+    points = [[0, 0, x] for x in range(3)]
+    pointcloud = Pointcloud(points)
+    assert str(pointcloud) == "Pointcloud(len(points)=3)"
+    assert repr(pointcloud) == "Pointcloud(points=[Point(x=0.0, y=0.0, z=0.0), Point(x=0.0, y=0.0, z=1.0), Point(x=0.0, y=0.0, z=2.0)])"
+
+
 def test_pointcloud__eq__():
     a = Pointcloud.from_bounds(10, 10, 10, 10)
     points = a.points[:]

tests/compas/geometry/test_polygon.py

@@ -52,6 +52,13 @@ def test_polygon_data():
         assert Polygon.validate_data(other.__data__)
 
 
+def test_polygon_str_repr():
+    points = [[0, 0, x] for x in range(3)]
+    polygon = Polygon(points)
+    assert str(polygon) == "Polygon(points=[Point(x=0.000, y=0.000, z=0.000), Point(x=0.000, y=0.000, z=1.000), Point(x=0.000, y=0.000, z=2.000)])"
+    assert repr(polygon) == "Polygon(points=[Point(x=0.0, y=0.0, z=0.0), Point(x=0.0, y=0.0, z=1.0), Point(x=0.0, y=0.0, z=2.0)])"
+
+
 def test_polygon__eq__():
     points1 = [[0, 0, x] for x in range(5)]
     polygon1 = Polygon(points1)

tests/compas/geometry/test_quaternion.py

@@ -64,6 +64,12 @@ def test_quaternion_data():
         assert Quaternion.validate_data(other.__data__)
 
 
+def test_quaternion_str_repr():
+    quaternion = Quaternion(0.5, 0.5, 0.5, 0.5)
+    assert str(quaternion) == "Quaternion(0.500, 0.500, 0.500, 0.500)"
+    assert repr(quaternion) == "Quaternion(0.5, 0.5, 0.5, 0.5)"
+
+
 # =============================================================================
 # Properties and Geometry
 # =============================================================================