Fixed world.cleanup() bug and extended test_connect_grids to test it.

Author: magicguy38

pathfinding/core/world.py

@@ -12,7 +12,7 @@ def __init__(self, grids: Dict[int, Grid]):
         self.dirty = False
 
     def cleanup(self):
-        for grid in self.grids:
+        for grid in self.grids.values():
             grid.cleanup()
 
     def neighbors(

test/test_connect_grids.py

@@ -25,6 +25,22 @@
     (0, 0, 1)
 ]
 
+PATH2 = [
+    # test same path a second time
+    (0, 0, 1),
+    (1, 0, 1),
+    (2, 0, 1),
+    (2, 1, 1),
+    (2, 2, 1),
+    (2, 2, 0),
+    (1, 2, 0),
+    (0, 2, 0),
+    (0, 1, 0),
+    (0, 0, 0),
+    (1, 0, 0),
+    (2, 0, 0)
+    # this ensures that world cleanup is working properly
+]
 
 def test_connect():
     level0 = [
@@ -52,4 +68,6 @@ def test_connect():
 
     finder = AStarFinder()
     path, _ = finder.find_path(grid0.node(2, 0), grid1.node(0, 0), world)
+    path2, _ = finder.find_path(grid1.node(0, 0), grid0.node(2, 0), world)
     assert [tuple(p) for p in path] == PATH
+    assert [tuple(p) for p in path2] == PATH2