def test_parser(self):
"""LineGrouper should return n non-blank lines at a time"""
good = [" \t >abc \n", "\t def\n", "\t\n", "\t >efg \n", "ghi"]
c = LineGrouper(2)
self.assertEqual(list(c(good)), [[">abc", "def"], [">efg", "ghi"]])
c = LineGrouper(1)
self.assertEqual(list(c(good)), [[">abc"], ["def"], [">efg"], ["ghi"]])
c = LineGrouper(4)
self.assertEqual(list(c(good)), [[">abc", "def", ">efg", "ghi"]])
# shouldn't work if not evenly divisible
c = LineGrouper(3)
self.assertRaises(RecordError, list, c(good))
def test_parser_ignore(self):
"""LineGrouper should skip lines to ignore."""
def never(line):
return False
def ignore_labels(line):
return (not line) or line.isspace() or line.startswith("#")
lines = ["abc", "\n", "1", "def", "#ignore", "2"]
self.assertEqual(list(LineGrouper(1)(lines)),
[["abc"], ["1"], ["def"], ["#ignore"], ["2"]])
self.assertEqual(list(LineGrouper(1, ignore=never)(lines)),
[[i.strip()] for i in lines])
self.assertEqual(
list(LineGrouper(2, ignore=ignore_labels)(lines)),
[["abc", "1"], ["def", "2"]],
)