class Test_Single_Player(TestCase):
@mock.patch('tictactoe.TicTacToe.input', create=True)
def test_comp_goes_after_player(self, mocked_input):
mocked_input.side_effect = ['a1']
self.game = TTT(width=3, first='player')
self.assertEqual(self.game.turn, self.game.Turn.PLAYER)
self.game.make_player_move()
self.assertEqual(self.game.turn, self.game.Turn.COMPUTER)
def tearDown(self):
del self.game
class Test_Custom_Game(TestCase):
def setUp(self):
self.game = TTT(5, 'computer')
def test_player_choice(self):
self.assertEqual(self.game.turn, self.game.Turn.COMPUTER)
def test_custom_game_width(self):
self.assertEqual(self.game.width, 5)
def test_advance_turn(self):
self.assertEqual(self.game.turn, self.game.Turn.COMPUTER)
self.game._advance_turn()
self.assertEqual(self.game.turn, self.game.Turn.PLAYER)
def tearDown(self):
del self.game
def main():
parser = argparse.ArgumentParser(description='Tic-Tac-Toe Game')
parser.add_argument('-w',
'--width',
default=3,
type=int,
help='Width & Height of the board')
parser.add_argument('-f',
'--first',
default='player',
choices=['player', 'computer', 'random'],
help='Who makes first move')
parser.add_argument('-m',
'--multi',
action='store_true',
help='Player vs Player')
args = parser.parse_args()
# Create an instance of TicTacToe
game = TicTacToe(width=args.width, first=args.first, multi=args.multi)
# Display game info and begin getting moves
game.show_game()
while True:
game.make_move()
game.show_game()
# Stop game if someone won or there's a tie
if game.get_winner() is not None:
break
# Display who won
win = game.get_winner()
if (win == game.Turn.PLAYER.value or win == game.Turn.COMPUTER.value):
print(f'Player {game.get_winner()} wins!')
else:
print("It's a tie. You both lose!")
def setUp(self):
self.game = TTT()
class Test_Default_Game(TestCase):
def setUp(self):
self.game = TTT()
def test_player_choice(self):
self.assertEqual(self.game.turn, self.game.Turn.PLAYER)
def test_default_width_3(self):
self.assertEqual(self.game.width, 3)
def test_advance_turn(self):
self.assertEqual(self.game.turn, self.game.Turn.PLAYER)
self.game._advance_turn()
self.assertEqual(self.game.turn, self.game.Turn.COMPUTER)
def test_hori_wins(self):
for row in range(self.game.width):
self.game._reset_board()
self.assertEqual(self.game.get_winner(), None)
for col in range(self.game.width):
self.game.board[row][col] = self.game.Turn.PLAYER.value
self.assertEqual(self.game.get_winner(),
self.game.Turn.PLAYER.value)
def test_vert_wins(self):
for col in range(self.game.width):
self.game._reset_board()
self.assertEqual(self.game.get_winner(), None)
for row in range(self.game.width):
self.game.board[row][col] = self.game.Turn.COMPUTER.value
self.assertEqual(self.game.get_winner(),
self.game.Turn.COMPUTER.value)
def test_diag_tl_br_win(self):
for row in range(self.game.width):
self.assertEqual(self.game.get_winner(), None)
for col in range(self.game.width):
if row == col:
self.game.board[row][col] = self.game.Turn.PLAYER.value
self.assertEqual(self.game.get_winner(), self.game.Turn.PLAYER.value)
def test_diag_bl_tr_win(self):
for row in range(self.game.width):
self.assertEqual(self.game.get_winner(), None)
for col in range(self.game.width):
if row + col == self.game.width - 1:
self.game.board[row][col] = self.game.Turn.COMPUTER.value
self.assertEqual(self.game.get_winner(), self.game.Turn.COMPUTER.value)
def test_tie_game(self):
symbs = (self.game.Turn.PLAYER.value, self.game.Turn.COMPUTER.value)
for row in range(self.game.width - 1):
for col in range(self.game.width):
self.game.board[row][col] = symbs[col % 2]
for col in range(self.game.width):
self.game.board[-1][col] = symbs[(col + 1) % 2]
self.assertEqual(self.game.get_winner(), 'T')
def tearDown(self):
del self.game
def test_convert__r0_c0(self):
spot = TTT.convert_row_col(0, 0)
self.assertEqual(spot, "A1")
def test_comp_goes_after_player(self, mocked_input):
mocked_input.side_effect = ['a1']
self.game = TTT(width=3, first='player')
self.assertEqual(self.game.turn, self.game.Turn.PLAYER)
self.game.make_player_move()
self.assertEqual(self.game.turn, self.game.Turn.COMPUTER)
def test_convert_r2_c2(self):
spot = TTT.convert_row_col(2, 2)
self.assertEqual(spot, "C3")
def test_convert_r2_c1(self):
spot = TTT.convert_row_col(2, 1)
self.assertEqual(spot, "B3")
def test_convert_r2_c0(self):
spot = TTT.convert_row_col(2, 0)
self.assertEqual(spot, "A3")
def test_convert_r1_c2(self):
spot = TTT.convert_row_col(1, 2)
self.assertEqual(spot, "C2")
def test_convert_r1_c1(self):
spot = TTT.convert_row_col(1, 1)
self.assertEqual(spot, "B2")
def test_convert_r1_c0(self):
spot = TTT.convert_row_col(1, 0)
self.assertEqual(spot, "A2")
def test_convert_r0_c2(self):
spot = TTT.convert_row_col(0, 2)
self.assertEqual(spot, "C1")
def setUp(self):
self.game = TTT(5, 'computer')
def test_convert_r0_c1(self):
spot = TTT.convert_row_col(0, 1)
self.assertEqual(spot, "B1")