Update test_space.py

Rbnstrn · Aug 5, 2019 · 092b922 · 092b922
1 parent f77e74c
commit 092b922
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_space.py b/tests/test_space.py
@@ -2,6 +2,7 @@
 
 import networkx as nx
 import numpy as np
+import pytest
 
 from mesa.space import ContinuousSpace
 from mesa.space import SingleGrid
@@ -249,7 +250,7 @@ def test_remove_middle(self):
 
 class TestSingleGrid(unittest.TestCase):
     def setUp(self):
-        self.space = SingleGrid(20, 20, False)
+        self.space = SingleGrid(50, 50, False)
         self.agents = []
         for i, pos in enumerate(TEST_AGENTS_GRID):
             a = MockAgent(i, None)
@@ -274,10 +275,11 @@ def test_remove_agent(self):
             assert self.space.grid[pos[0]][pos[1]] is None
 
     def test_empty_cells(self):
-        while self.space.exists_empty_cells():
-            pos = self.space.find_empty()
-            a = MockAgent(-99, pos)
-            self.space.position_agent(a, x=pos[0], y=pos[1])
+        if self.space.exists_empty_cells():
+            pytest.deprecated_call(self.space.find_empty)
+            for i, pos in enumerate(list(self.space.empties)):
+                a = MockAgent(-i, pos)
+                self.space.position_agent(a, x=pos[0], y=pos[1])
         assert self.space.find_empty() is None
         with self.assertRaises(Exception):
             self.space.move_to_empty(a)