def test_get_radius_0(self): s = Space((6,6,3)) p = PathManager(s) for x in range(6): for y in range(6): s[(x,y,0)] = 1 self.assertEqual(p.within_radius((2,2,0),0), set([(2,2,0)]))
def test_get_radius_2(self): s = Space((6,6,3)) p = PathManager(s) for x in range(6): for y in range(6): s[(x,y,0)] = 1 self.assertEqual(p.within_radius((2,2,0),2), set([(2,1,0),(3,1,0),(3,2,0),(2,3,0),(1,2,0), (1,1,0),(2,2,0),(2,0,0),(3,0,0),(4,1,0), (4,2,0),(4,3,0),(3,3,0),(2,4,0),(1,3,0), (0,3,0),(0,2,0),(0,1,0),(1,0,0)]))