def test_get_neighbor_cells(self):
t = StateSuccessorFinder(States.state0)
self.assertEqual(t.get_neighbor_cells((0, 0)), {
"u": None,
"r": (0, 1),
"d": (1, 0),
"l": None
})
self.assertEqual(t.get_neighbor_cells((1, 0)), {
"u": (0, 0),
"r": (1, 1),
"d": (2, 0),
"l": None
})
self.assertEqual(t.get_neighbor_cells((M - 1, N - 1)), {
"u": (M - 2, N - 1),
"r": None,
"d": None,
"l": (M - 1, N - 2)
})
self.assertEqual(t.get_neighbor_cells((1, 2)), {
"u": (0, 2),
"r": (1, 3),
"d": (2, 2),
"l": (1, 1)
})
def test_get_neighbor_cells(self):
t = StateSuccessorFinder(States.state0)
self.assertEqual(t.get_neighbor_cells((0, 0)),
{"u": None, "r": (0, 1), "d": (1, 0), "l": None})
self.assertEqual(t.get_neighbor_cells((1, 0)),
{"u": (0, 0), "r": (1, 1), "d": (2, 0), "l": None})
self.assertEqual(t.get_neighbor_cells((M - 1, N - 1)),
{"u": (M - 2, N - 1), "r": None, "d": None, "l": (M - 1, N - 2)})
self.assertEqual(t.get_neighbor_cells((1, 2)),
{"u": (0, 2), "r": (1, 3), "d": (2, 2), "l": (1, 1)})
