def test_game_win():
small_board = Board(2, 2)
small_board.set_value((0, 0), 'boom')
small_board.set_value((0, 1), 'boom')
small_board.set_value((1, 0), 'boom')
small_board.fill_board()
small_board.dig((1, 1))
assert small_board.game_over is True
assert small_board.game_won is True
def test_rupoor():
new = Board(2, 2)
new.set_value((0, 0), 'rupoor')
new.fill_board()
new.dig((0, 1))
assert new.bombs == 1
assert new.dig((0, 0)) == 'rupoor'
assert new.score == 0
assert new.values[1][1] == 'blue'
assert new.bombs == 0
def test_game_start():
new = Board()
new.game_start(booms=5, rupoors=4)
assert new.bombs == 9
values_list = []
for row in new.values:
values_list += row
for col in row:
assert col is not None
assert values_list.count('boom') == 5
assert values_list.count('rupoor') == 4
new.dig((0, 0))
new.game_start()
assert not new.game_over