def test_open_adjacent(self):
s = Space((4,4,3))
for x in range(4):
for y in range(4):
s[(x,y,0)] = 1
s[(1,0,1)] = s[(1,0,2)] = 1
s[(2,2,1)] = s[(2,2,2)] = 1
p = PathManager(s)
t = (2,1,1)
a = set([ (2,0,1),
(3,0,1),
(1,1,1), (3,1,1)])
self.assertEqual(p.open_adjacent(t), a)
def test_open_adjacent_step_down_blocked(self):
s = Space((4,4,3))
for x in range(4):
for y in range(4):
s[(x,y,0)] = 1
s[(x,y,1)] = 1
s[(1,0,1)] = 0
s[(1,0,2)] = 1
s[(2,2,1)] = 0
p = PathManager(s)
t = (2,1,2)
a = set([ (2,0,2),
(3,0,2),
(1,1,2), (3,1,2),
(2,2,1)])
self.assertEqual(p.open_adjacent(t), a)
