def test_ask_cells_for_tied_game(self): selected_cells = [1, 5, 9, 2, 8, 7, 3, 6, 4] game = Game(lambda _: selected_cells.pop(), self.print_function) game.board.clean_marks() game.ask_cells() self.assertFalse(game.is_won()) expected_printed_messages = ' The game is over and this is a tie!!' messages = self.returned_string.split('\n') win_message = messages[len(messages) - 2] self.assertEqual(win_message, expected_printed_messages)
def test_ask_cells_for_won_game(self): game = Game(lambda n: 3, self.print_function) game.board.clean_marks() game.board.mark_cell(1, 1) game.board.mark_cell(2, 1) game.ask_cells() self.assertTrue(game.is_won()) expected_printed_messages = ' Player 1 wins the game!!' messages = self.returned_string.split('\n') win_message = messages[len(messages) - 2] self.assertEqual(win_message, expected_printed_messages)