def test_get_cells_symbol_zero():
""" Check there is zero match
"""
array = [[0, 1, 1, 0, 1, 1, 0], [0, 1, 1, 0, 1, 1,
0], [0, 1, 1, 0, 1, 1, 0],
[0, 1, 1, 0, 1, 1, 0], [1, 1, 1, 0, 1, 1, 0],
[1, 1, 1, 1, 1, 1, 0]]
board = Board()
board.board = array
symb_a = 3
symb_b = 2
# positions of the symbol 3 and 2
pos_a = []
pos_b = []
# compare all cells of each symbol with the respective array
assert (board.get_cells_symbol(symb_a) == pos_a)
assert (board.get_cells_symbol(symb_b) == pos_b)
def test_get_cells_symbol_one_or_more():
""" Check there is at least one or more match
"""
array = [[0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0,
0], [0, 1, 0, 0, 0, 0, 0],
[2, 2, 1, 0, 2, 0, 0], [1, 1, 2, 2, 1, 1, 0],
[2, 1, 2, 1, 1, 2, 2]]
board = Board()
board.board = array
symb_a = 1
symb_b = 2
# positions of the symbol 1 and 2
pos_a = [(2, 1), (3, 2), (4, 0), (4, 1), (4, 4), (4, 5), (5, 1), (5, 3),
(5, 4)]
pos_b = [(3, 0), (3, 1), (3, 4), (4, 2), (4, 3), (5, 0), (5, 2), (5, 5),
(5, 6)]
# compare all cells of each symbol with the respective array
assert (board.get_cells_symbol(symb_a) == pos_a)
assert (board.get_cells_symbol(symb_b) == pos_b)
