def test_5p_extension(self): a = read_alignment("""> target AAAAAAAAAAGCGGAUUUALCUCAG > template ----------GCGGAUUUALCUCAG """) m = RnaModel(None, a, data_type='file', data=MINI_TEMPLATE) m.add_missing_5p() self.assertEqual(m.get_sequence(), a.target_seq)
def test_long_5p_extension(self): a = read_alignment("""> target AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAGCGGAUUUALCUCAG > template ----------------------------------------GCGGAUUUALCUCAG """) # first number == 1 should not work m = RnaModel(None, a, data_type='file', data=MINI_TEMPLATE) self.assertRaises(RenumeratorError, m.add_missing_5p) # first number == 100 should work m = RnaModel(None, a, data_type='file', data=MINI_TEMPLATE) renumber_chain(m, 100) m.add_missing_5p() self.assertEqual(m.get_sequence(), a.target_seq)