Fix test_visualization param bug

projectmesa · Jul 7, 2017 · 364ad70 · 364ad70
1 parent 92bfcf3
commit 364ad70
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mesa/batchrunner.py b/mesa/batchrunner.py
@@ -17,10 +17,11 @@ def combinations(*items):
     """
     A small fix to handle dictionary type parameters in cartesian product.
     """
-    prepared = [(item,) if isinstance(item, Mapping) else item
+    prepared = [(item,) if isinstance(item, collections.Mapping) else item
                 for item in items]
     yield from (param for param in product(*prepared))
 
+
 class VariableParameterError(TypeError):
     MESSAGE = ('variable_parameters must map a name to a sequence of values. '
             'These parameters were given with non-sequence values: {}')

diff --git a/tests/test_visualization.py b/tests/test_visualization.py
@@ -24,7 +24,7 @@ def __init__(self, width, height, key1=103, key2=104):
         self.grid = Grid(width, height, torus=True)
 
         for (c, x, y) in self.grid.coord_iter():
-            a = MockAgent(x + y * 100, x * y * 3)
+            a = MockAgent(x + y * 100, self, x * y * 3)
             self.grid.place_agent(a, (x, y))
             self.schedule.add(a)