class TestBoard(unittest.TestCase): def setUp(self): self.board = Board(10,10) def test_initialise_collection(self): self.board.initialise_collection() self.assertEqual(len(self.board.collection), 100) def test_get_neighbours(self): for i in range (100): self.board.collection.append(Cell(False))