Refactored paint_connect() and path_connect() to receive and return FlatCAMRTreeStorage objects. Updated unittests acordingly.

tests/test_pathconnect.py

@@ -7,9 +7,20 @@ from camlib import *
 from random import random
 
 
+def mkstorage(paths):
+    def get_pts(o):
+        return [o.coords[0], o.coords[-1]]
+    storage = FlatCAMRTreeStorage()
+    storage.get_points = get_pts
+    for p in paths:
+        storage.insert(p)
+    return storage
+
+
 class PathConnectTest1(unittest.TestCase):
 
     def setUp(self):
+        print "PathConnectTest1.setUp()"
         pass
 
     def test_simple_connect(self):
@@ -18,8 +29,9 @@ class PathConnectTest1(unittest.TestCase):
             LineString([[1, 1], [2, 1]])
         ]
 
-        result = Geometry.path_connect(paths)
+        result = Geometry.path_connect(mkstorage(paths))
 
+        result = list(result.get_objects())
         self.assertEqual(len(result), 1)
         self.assertTrue(result[0].equals(LineString([[0, 0], [1, 1], [2, 1]])))
 
@@ -30,8 +42,9 @@ class PathConnectTest1(unittest.TestCase):
             LineString([[-0.5, 0.5], [0.5, 0]])
         ]
 
-        result = Geometry.path_connect(paths)
+        result = Geometry.path_connect(mkstorage(paths))
 
+        result = list(result.get_objects())
         self.assertEqual(len(result), 2)
         matches = [p for p in result if p.equals(LineString([[0, 0], [1, 1], [2, 1]]))]
         self.assertEqual(len(matches), 1)
@@ -46,8 +59,9 @@ class PathConnectTest1(unittest.TestCase):
                 LineString([[1 + offset_x, 1 + offset_y], [2 + offset_x, 1 + offset_y]])
             ]
 
-            result = Geometry.path_connect(paths)
+            result = Geometry.path_connect(mkstorage(paths))
 
+            result = list(result.get_objects())
             self.assertEqual(len(result), 1)
             self.assertTrue(result[0].equals(LineString([[0 + offset_x, 0 + offset_y],
                                                          [1 + offset_x, 1 + offset_y],
@@ -63,8 +77,9 @@ class PathConnectTest1(unittest.TestCase):
             LinearRing([[1, 1], [2, 2], [1, 3], [0, 2]])
         ]
 
-        result = Geometry.path_connect(paths)
+        result = Geometry.path_connect(mkstorage(paths))
 
+        result = list(result.get_objects())
         self.assertEqual(len(result), 2)
         matches = [p for p in result if p.equals(LineString([[0, 0], [1, 1], [2, 1]]))]
         self.assertEqual(len(matches), 1)