fix custom magic methods and add tests

Author: beverlylytle

CHANGELOG.md

@@ -19,6 +19,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 * Fixed bug where mimic joints were considered configurable.
 * Fixed bug where `!=` gave incorrect results in Rhino for some compas objects.
 * Fixed bug where `compas_rhino.BaseArtist.redraw` did not trigger a redraw.
+* Fixed minor bugs in `compas.geometry.Polyline` and `compas.geometry.Polygon`.
+* Fixed very minor bugs in `compas.geometry.Frame` and `compas.geometry.Quaternion`.
 
 ### Removed
 

src/compas/geometry/primitives/frame.py

@@ -180,6 +180,8 @@ def __iter__(self):
         return iter([self.point, self.xaxis, self.yaxis])
 
     def __eq__(self, other, tol=1e-05):
+        if not hasattr(other, '__iter__') or not hasattr(other, '__len__') or len(self) != len(other):
+            return False
         for v1, v2 in zip(self, other):
             for a, b in zip(v1, v2):
                 if math.fabs(a - b) > tol:

src/compas/geometry/primitives/polygon.py

@@ -147,7 +147,7 @@ def area(self):
     # ==========================================================================
 
     def __repr__(self):
-        return "Polygon({})".format(", ".join(["{}".format(point) for point in self.points]))
+        return "Polygon([{}])".format(", ".join(["{}".format(point) for point in self.points]))
 
     def __len__(self):
         return len(self.points)
@@ -156,12 +156,15 @@ def __getitem__(self, key):
         return self.points[key]
 
     def __setitem__(self, key, value):
-        self.points[key] = value
+        self.points[key] = Point(*value)
+        self._lines = None
 
     def __iter__(self):
         return iter(self.points)
 
     def __eq__(self, other):
+        if not hasattr(other, '__iter__') or not hasattr(other, '__len__') or len(self) != len(other):
+            return False
         return all(a == b for a, b in zip(self, other))
 
     # ==========================================================================

src/compas/geometry/primitives/polyline.py

@@ -109,7 +109,7 @@ def length(self):
     # ==========================================================================
 
     def __repr__(self):
-        return "Polyline({})".format(", ".join(["{}".format(point) for point in self.points]))
+        return "Polyline([{}])".format(", ".join(["{}".format(point) for point in self.points]))
 
     def __len__(self):
         return len(self.points)
@@ -118,12 +118,15 @@ def __getitem__(self, key):
         return self.points[key]
 
     def __setitem__(self, key, value):
-        self.points[key] = value
+        self.points[key] = Point(*value)
+        self._lines = None
 
     def __iter__(self):
         return iter(self.points)
 
     def __eq__(self, other):
+        if not hasattr(other, '__iter__') or not hasattr(other, '__len__') or len(self) != len(other):
+            return False
         return all(a == b for a, b in zip(self, other))
 
     # ==========================================================================

src/compas/geometry/primitives/quaternion.py

@@ -217,6 +217,8 @@ def __setitem__(self, key, value):
         raise KeyError
 
     def __eq__(self, other, tol=1e-05):
+        if not hasattr(other, '__iter__') or not hasattr(other, '__len__') or len(self) != len(other):
+            return False
         for v1, v2 in zip(self, other):
             if math.fabs(v1 - v2) > tol:
                 return False
@@ -225,6 +227,9 @@ def __eq__(self, other, tol=1e-05):
     def __iter__(self):
         return iter(self.wxyz)
 
+    def __len__(self):
+        return 4
+
     def __repr__(self):
         return 'Quaternion({:.{prec}f}, {:.{prec}f}, {:.{prec}f}, {:.{prec}f})'.format(self.w, self.x, self.y, self.z, prec=3)
 

tests/compas/geometry/test_polygon.py

@@ -0,0 +1,55 @@
+import pytest
+
+from compas.geometry import Point
+from compas.geometry import Polygon
+from compas.utilities import pairwise
+
+
+def test_polygon():
+    points = [[0, 0, x] for x in range(5)]
+    polygon = Polygon(points)
+    assert polygon.points == points
+    assert polygon.lines == [(a, b) for a, b in pairwise(points + points[:1])]
+
+
+def test_equality():
+    points1 = [[0, 0, x] for x in range(5)]
+    polygon1 = Polygon(points1)
+    points2 = [[0, 0, x] for x in range(6)]
+    polygon2 = Polygon(points2)
+    points3 = [[0, 0, x] for x in range(5)] + [[0, 0, 0]]
+    polygon3 = Polygon(points3)
+    assert polygon1 == polygon1
+    assert polygon1 == points1
+    assert points1 == polygon1
+    assert polygon1 != polygon2
+    assert polygon2 != polygon1
+    assert polygon1 != points2
+    assert points2 != polygon1
+    assert polygon1 != 1
+    assert polygon1 == polygon3
+
+
+def test___repr__():
+    points = [[0, 0, x] for x in range(5)]
+    polygon = Polygon(points)
+    assert polygon == eval(repr(polygon))
+
+
+def test___getitem__():
+    points = [[0, 0, x] for x in range(5)]
+    polygon = Polygon(points)
+    for x in range(5):
+        assert polygon[x] == [0, 0, x]
+    with pytest.raises(IndexError):
+        polygon[6] = [0, 0, 6]
+
+
+def test___setitem__():
+    points = [[0, 0, x] for x in range(5)]
+    polygon = Polygon(points)
+    point = [1, 1, 4]
+    polygon[4] = point
+    assert polygon[4] == point
+    assert isinstance(polygon[4], Point)
+    assert polygon.lines[-2].end == point

tests/compas/geometry/test_polyline.py

@@ -1,7 +1,56 @@
-import compas
-import pytest
-from compas.geometry import Polyline, Point
 import math
+import pytest
+
+from compas.geometry import Point
+from compas.geometry import Polyline
+from compas.utilities import pairwise
+
+
+def test_polyline():
+    points = [[0, 0, x] for x in range(5)]
+    polyline = Polyline(points)
+    assert polyline.points == points
+    assert polyline.lines == [(a, b) for a, b in pairwise(points)]
+
+
+def test_equality():
+    points1 = [[0, 0, x] for x in range(5)]
+    polyline1 = Polyline(points1)
+    points2 = [[0, 0, x] for x in range(6)]
+    polyline2 = Polyline(points2)
+    assert polyline1 == polyline1
+    assert polyline1 == points1
+    assert points1 == polyline1
+    assert polyline1 != polyline2
+    assert polyline2 != polyline1
+    assert polyline1 != points2
+    assert points2 != polyline1
+    assert polyline1 != 1
+
+
+def test___repr__():
+    points = [[0, 0, x] for x in range(5)]
+    polyline = Polyline(points)
+    assert polyline == eval(repr(polyline))
+
+
+def test___getitem__():
+    points = [[0, 0, x] for x in range(5)]
+    polyline = Polyline(points)
+    for x in range(5):
+        assert polyline[x] == [0, 0, x]
+    with pytest.raises(IndexError):
+        polyline[6] = [0, 0, 6]
+
+
+def test___setitem__():
+    points = [[0, 0, x] for x in range(5)]
+    polyline = Polyline(points)
+    point = [1, 1, 4]
+    polyline[4] = point
+    assert polyline[4] == point
+    assert isinstance(polyline[4], Point)
+    assert polyline.lines[-1].end == point
 
 
 @pytest.mark.parametrize('coords,expected', [