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"))
