def test3Generations(self): competitors = ['TitForTat', 'Naive'] generational = Generational('PrisonerMatrix', competitors, 2) #override initial population generational._queue = competitors generational.start() self.assertEqual(generational._queue.count('TitForTat'), 2) self.assertEqual(generational._queue.count('Naive'), 4) generational._next_generation() self.assertEqual(generational._queue.count('TitForTat'), 4) self.assertEqual(generational._queue.count('Naive'), 4)