def test_multi(self): r = list(transform.drop_columns(self.sequences, [slice(0, 1), slice(2, None)])) self.assertEqual([i.id for i in self.sequences], [i.id for i in r]) self.assertEqual(['A', '-', 'A'], [str(i.seq) for i in r])
def test_basic(self): r = list(transform.drop_columns(self.sequences, [slice(1, None)])) self.assertEqual([i.id for i in self.sequences], [i.id for i in r]) self.assertEqual(['A', 'A', '-'], [str(i.seq) for i in r])