def test_bases(self): """CodonUsage bases should count bases correctly""" u = CodonUsage('UUUCCCUAGCCCGGGAA') b = u.bases() self.assertEqual(b, BaseUsage('UUUCCCUAGCCCGGGAA')) #purge_unwanted should get rid of bad codons b = u.bases(purge_unwanted=True) self.assertEqual(b, BaseUsage('UUUCCCCCCGGG'))