def test_get_column_exist():
""" 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
col_0 = [0, 0, 0, 0, 1, 1]
col_1 = [1, 1, 1, 1, 1, 1]
col_2 = [1, 1, 1, 1, 1, 1]
col_3 = [0, 0, 0, 0, 0, 1]
col_4 = [1, 1, 1, 1, 1, 1]
col_5 = [1, 1, 1, 1, 1, 1]
col_6 = [0, 0, 0, 0, 0, 0]
assert (board.get_column(0) == col_0)
assert (board.get_column(1) == col_1)
assert (board.get_column(2) == col_2)
assert (board.get_column(3) == col_3)
assert (board.get_column(4) == col_4)
assert (board.get_column(5) == col_5)
assert (board.get_column(6) == col_6)
def test_get_column_not_exist():
""" 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
col_7 = []
col_8 = []
col_20 = []
assert (board.get_column(7) == col_7)
assert (board.get_column(8) == col_8)
assert (board.get_column(20) == col_20)
