def test_trim_seqs_notrim_outofbounds(self):
seqs = [("seq1", "tagggcaagactccatggtatga"),
("seq2", "cggaggcgagatgcgtggta"),
("seq3", "tactagcaagattcctggtaaagga"),
("seq4", "aggatgcgagatgcgtg"),
("seq5", "gagtgcgagatgcgtggtgagg"),
("seq6", "ggatgcgagatgcgtggtgatt"),
("seq7", "agggcgagattcctagtgga--")]
with self.assertRaises(ValueError):
list(trim_seqs(seqs, -2, 0))
def test_trim_seqs_notrim(self):
seqs = [("seq1", "tagggcaagactccatggtatga"),
("seq2", "cggaggcgagatgcgtggta"),
("seq3", "tactagcaagattcctggtaaagga"),
("seq4", "aggatgcgagatgcgtg"),
("seq5", "gagtgcgagatgcgtggtgagg"),
("seq6", "ggatgcgagatgcgtggtgatt"),
("seq7", "agggcgagattcctagtgga--")]
obs = trim_seqs(seqs, -1, 0)
self.assertEqual(list(obs), seqs)
def test_trim_seqs_left(self):
seqs = [("seq1", "tagggcaagactccatggtatga"),
("seq2", "cggaggcgagatgcgtggta"),
("seq3", "tactagcaagattcctggtaaagga"),
("seq4", "aggatgcgagatgcgtg"),
("seq5", "gagtgcgagatgcgtggtgagg"),
("seq6", "ggatgcgagatgcgtggtgatt"),
("seq7", "agggcgagattcctagtgga--")]
obs = trim_seqs(seqs, 20, 5)
self.assertTrue(isinstance(obs, GeneratorType))
exp = [("seq1", "caagactccatggta"), ("seq2", "gcgagatgcgtggta"),
("seq3", "gcaagattcctggta"), ("seq5", "cgagatgcgtggtga"),
("seq6", "cgagatgcgtggtga"), ("seq7", "gagattcctagtgga")]
self.assertEqual(list(obs), exp)
