class CreateBoardTest(TestCase): def setUp(self): self.create_board = CreateBoard(BOARD_ROWS, BOARD_COLUMNS, TOTAL_MINES) self.board = self.create_board.get_board() def test_create_board(self): self.failUnlessEqual(len(self.board), BOARD_ROWS) self.failUnlessEqual(len(self.board[0]), BOARD_COLUMNS) def test_created_10_mines(self): mines = 0 for row in range(BOARD_ROWS): for column in range(BOARD_COLUMNS): mines += self.board[row][column] self.failUnlessEqual(mines, TOTAL_MINES)
def setUp(self): self.create_board = CreateBoard(BOARD_ROWS, BOARD_COLUMNS, TOTAL_MINES) self.board = self.create_board.get_board()