def test_find_gene(self): perc = Chromosome(1, 2) perc.add_gene("HTT") perc.add_gene("MIT") self.assertEquals(True, perc.find_gene("HTT")) self.assertEquals(True, perc.find_gene("MIT")) self.assertEquals(False, perc.find_gene("HEH"))
def test_add_gene(self): self.assertEquals(True, Chromosome(1, 1).add_gene("HTT")) self.assertEquals(True, Chromosome(1, 1).add_gene(Gene("HTT"))) # Add a gene to the persistent chromosome perc = Chromosome(1, 1) perc.add_gene("HTT") self.assertEquals(1, len(perc.genes)) self.assertEquals("HTT", perc.genes[0].name) self.assertEquals(1, len(perc.locus)) self.assertEquals("HTT", perc.locus[0].name) # Exceptions self.assertRaises(IndexError, lambda: perc.add_gene("RSX"))