def test_chunk_lines_size_n(self):
from xin import chunked_lines
def woods_excerpt(start, end):
return '\n'.join(woods_txt.split('\n')[start:end]) + '\n'
def new_woods_file():
return StringIO(woods_txt)
self.assertEquals(woods_excerpt(0, 5), list(chunked_lines(new_woods_file(), 5))[0])
self.assertEquals(woods_excerpt(2, 4), list(chunked_lines(new_woods_file(), 2))[1])
def test_chunk_lines_default(self):
from xin import chunked_lines
num_lines = len(woods_txt.split('\n')) - 1
self.assertEquals(num_lines, len(list(chunked_lines(self.woods_file))))
