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))))