def test_mask_whole(self): masks = [slice(0, 200)] actual = list(transform.multi_mask_sequences(self.sequences, masks)) self.assertEqual(len(self.sequences), len(actual)) for e, a in zip(self.sequences, actual): self.assertEqual(e.id, a.id) self.assertEqual('-'*len(e), str(a.seq))
def test_mask(self): masks = [slice(1, 2)] actual = list(transform.multi_mask_sequences(self.sequences, masks)) self.assertEqual(len(self.sequences), len(actual)) self.assertEqual(['A-A', 'B-B', 'D-DD', 'E-E'], [str(a.seq) for a in actual])