class TestRows(unittest.TestCase):
def setUp(self):
self.game = Board()
def test_row1_player1(self):
self.game.board = [['X', 'X', 'X'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(self.game.check_rows(), 1)
def test_row1_player2(self):
self.game.board = [['O', 'O', 'O'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(self.game.check_rows(), 2)
def test_row2_player1(self):
self.game.board = [['_', '_', '_'], ['X', 'X', 'X'], ['_', '_', '_']]
self.assertEqual(self.game.check_rows(), 1)
def test_row2_player2(self):
self.game.board = [['_', '_', '_'], ['O', 'O', 'O'], ['_', '_', '_']]
self.assertEqual(self.game.check_rows(), 2)
def test_row3_player1(self):
self.game.board = [['_', '_', '_'], ['_', '_', '_'], ['X', 'X', 'X']]
self.assertEqual(self.game.check_rows(), 1)
def test_row3_player2(self):
self.game.board = [['_', '_', '_'], ['_', '_', '_'], ['O', 'O', 'O']]
self.assertEqual(self.game.check_rows(), 2)
def test_empty(self):
self.game.board = [['_', '_', '_'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(self.game.check_rows(), 0)
def test_no_rows(self):
self.game.board = [['X', '_', 'O'], ['_', 'X', 'O'], ['_', 'X', 'O']]
self.assertEqual(self.game.check_rows(), 0)
def test_check_rows_no_winner(self):
myBoard = Board()
myBoard.board = [['_', '_', 'O'], ['X', '_', '_'], ['X', 'O', 'O']]
self.assertEqual(myBoard.check_rows(), 0)
def test_check_rows_O_in_row_2(self):
myBoard = Board()
myBoard.board = [['_', '_', '_'], ['_', '_', '_'], ['O', 'O', 'O']]
self.assertEqual(myBoard.check_rows(), 2)
def test_check_rows_X_in_row_0(self):
myBoard = Board()
myBoard.board = [['X', 'X', 'X'], ['_', '_', '_'], ['_', '_', '_']]
self.assertEqual(myBoard.check_rows(), 1)
