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)