def test_prev_position(self): self.assertEqual( set(ricochet.prev_position(board, set(), (1, 1))), {(1, 0)}, ) self.assertEqual( set(ricochet.prev_position(board, set(), (0, 0))), {(0,1),(0,2),(0,3),(0,4),(1,0),(2,0),(3,0),(4,0)}, ) self.assertEqual( set(ricochet.prev_position(board, {(1, 0)}, (0, 0))), {(0,1),(0,2),(0,3),(0,4)}, ) self.assertEqual( set(ricochet.prev_position(board, {(2, 0)}, (0, 0))), {(0,1),(0,2),(0,3),(0,4),(1,0)}, )
def neighbour(coord): return ricochet.prev_position(board, set(), coord)