class TestPopulationBitstring(unittest.TestCase):
def setUp(self):
bit0 = Bitstring("0000")
bit1 = Bitstring("1111")
col0 = ColicinBit(bit0)
col1 = ColicinBit(bit1)
org = Organism([col0], [ImmunityBit(bit0, 2)])
self.pop = Population([org for _ in range(5)])
def test_replicate(self):
self.assertEqual(len(self.pop), 5)
self.pop.replicate()
self.assertEqual(len(self.pop), 10)
def test_remove_self_toxic(self):
self.assertEqual(len(self.pop), 5)
self.pop.remove_self_toxic()
self.assertEqual(len(self.pop), 5)
def test_cull_by_iterative_colicin(self):
self.assertEqual(len(self.pop), 5)
self.pop.cull_by_iterative_colicin()
self.assertEqual(len(self.pop), 5)
def test_advance_generation(self):
self.pop.advance_generation()
def test_sample_with_replacement(self):
self.pop.sample_with_replacement()
class TestPopulation(unittest.TestCase):
def setUp(self):
self.col = Colicin(0)
self.imm = Immunity(-5, 5)
self.org = Organism([self.col], [self.imm])
self.pop = Population([self.org.duplicate() for _ in range(10)])
def test_len(self):
self.assertEqual(len(self.pop), 10)
def test_replicate(self):
self.pop.replicate()
self.assertEqual(self.pop.pop.count(self.org), 10)
self.assertEqual(len(self.pop), 20)
def test_iter(self):
for org in self.pop:
self.assertEqual(org, self.org)
else:
return
self.fail()
def test_remove_self_toxic(self):
org_toxic = Organism([Colicin(0)], [Immunity(10, 5)])
org_not_toxic = Organism([Colicin(0)], [Immunity(0, 5)])
self.pop.pop = [org_not_toxic, org_toxic]
self.assertEqual(len(self.pop), 2)
self.pop.remove_self_toxic()
self.assertEqual(len(self.pop), 1)
self.assertSequenceEqual(self.pop.pop, [org_not_toxic])
def test_cull_by_all_colicins(self):
org_winner = Organism([Colicin(5)], [Immunity(0, 5)])
self.pop.pop.append(org_winner)
self.pop.cull_by_all_colicins()
self.assertSequenceEqual(self.pop.pop, [org_winner])
def test_cull_by_all_colicins_extinction(self):
org_other = Organism([Colicin(5)], [Immunity(10, 5)])
self.pop.pop.append(org_other)
self.pop.cull_by_all_colicins()
self.assertEqual(len(self.pop), 0)
def test_cull_by_single_colicin(self):
org_other = Organism([Colicin(100)], [Immunity(100, 5)])
self.pop.pop.extend(org_other for _ in range(10))
self.pop.cull_by_single_colicin()
self.assertEqual(len(self.pop), 10)
def test_cull_by_single_colicin_extinction(self):
self.pop.cull_by_single_colicin(Colicin(5))
self.assertEqual(len(self.pop), 0)
def test_cull_by_iterative_colicin(self):
org_winner = Organism([Colicin(5)], [Immunity(0, 5)])
self.pop.pop.append(org_winner)
self.pop.cull_by_iterative_colicin()
self.assertSequenceEqual(self.pop.pop, [org_winner])
def test_cull_by_iterative_colicin_extinction(self):
org_other = Organism([Colicin(5)], [Immunity(10, 5)])
self.pop.pop.extend(org_other for _ in range(10))
self.pop.cull_by_iterative_colicin()
self.assertEqual(len(self.pop), 10)
def test_colicins_produced(self):
colicins = set(self.pop.colicins_produced())
self.assertSetEqual({self.col}, colicins)
other_colicin = Colicin(5)
self.pop.pop.append(Organism([other_colicin], [Immunity(0, 5)]))
colicins = set(self.pop.colicins_produced())
self.assertSetEqual({self.col, other_colicin}, colicins)
def test_sample_with_replacement(self):
self.pop.pop.extend(self.org for _ in range(20))
self.assertEqual(len(self.pop), 30)
self.pop.sample_with_replacement()
self.assertEqual(len(self.pop), 10)
self.pop.pop = self.pop.pop[::2]
self.assertEqual(len(self.pop), 5)
self.pop.sample_with_replacement()
self.assertEqual(len(self.pop), 10)
self.assertEqual(self.pop.carrying_capacity, 10)
def test_advance_generation(self):
self.pop.advance_generation()
self.assertEqual(len(self.pop), 10)
