def test_zero_zero(self):
con = (0, 0)
neighbors = [(-1, -1),
(-1, 0),
(-1, 1),
(0, -1),
(0, 1),
(1, -1),
(1, 0),
(1, 1)]
assert set(main.get_neighbors(con)) == set(neighbors)
def test_positive_result(self):
con = (5, 5)
neighbors = [(4, 4),
(4, 5),
(4, 6),
(5, 4),
(5, 6),
(6, 4),
(6, 5),
(6, 6)]
assert set(main.get_neighbors(con)) == set(neighbors)
def test_zero_positive(self):
con = (0, 2)
neighbors = [(-1, 1),
(-1, 2),
(-1, 3),
(0, 1),
(0, 3),
(1, 1),
(1, 2),
(1, 3)]
assert main.get_neighbors(con) == neighbors
