def test_create_starts_and_stops(self):
gene = Gene()
mrna = Mock()
gene.children = {'mrna':[mrna]}
gene.create_starts_and_stops('ATGC')
mrna.create_starts_and_stops.assertCalledWith('ATGC')
def test_get_cds_length(self):
gene = Gene()
mrna = Mock()
cds = Mock()
cds.length = Mock(return_value=42)
mrna.get_cds = Mock(return_value=cds)
gene.children = {'mrna':[mrna]}
self.assertEquals(gene.get_cds_length(), 42)
def test_match_cds_and_exon_end(self):
gene = Gene()
mrna = Mock()
gene.children = {'mrna':[mrna]}
gene.match_cds_and_exon_end()
mrna.match_cds_and_exon_end.assertCalled()