Implement Edge.close

Author: adam-urbanczyk

cadquery/occ_impl/shapes.py

@@ -1386,6 +1386,20 @@ def _geomAdaptorH(self) -> Tuple[BRepAdaptor_Curve, BRepAdaptor_HCurve]:
 
         return curve, BRepAdaptor_HCurve(curve)
 
+    def close(self) -> Union["Edge", "Wire"]:
+        """
+        Close an Edge
+        """
+        rv: Union[Wire, Edge]
+
+        if not self.IsClosed():
+            e = Edge.makeLine(self.endPoint(), self.startPoint())
+            rv = Wire.assembleEdges((self, e))
+        else:
+            rv = self
+
+        return rv
+
     @classmethod
     def makeCircle(
         cls: Type["Edge"],

tests/test_cadquery.py

@@ -4410,6 +4410,14 @@ def testSplineApprox(self):
         with raises(ValueError):
             e4 = Edge.makeSplineApprox(pts, 1e-6, maxDeg=3, smoothing=(1, 1, 1.0))
 
+        pts_closed = pts + [pts[0]]
+
+        e3 = Edge.makeSplineApprox(pts_closed)
+        w = Edge.makeSplineApprox(pts).close()
+
+        self.assertTrue(e3.IsClosed())
+        self.assertTrue(w.IsClosed())
+
         N = 40
         T = 20
         A = 5