def test_change_limit_then_chunk():
data = "foo\n" * 200
m = MoreLimiter(1)
chunks = m.chunk(data)
m.change_limit(10)
for _ in xrange(20):
assert chunks.next() == ("foo\n" * 10)
def test_change_limit():
data = "foo\n" * 200
m = MoreLimiter(1)
m.change_limit(10)
chunks = m.chunk(data)
for _ in xrange(20):
res = chunks.next()
expected = ("foo\n" * 10)
print repr(expected)
assert res == expected
assert chunks.next() == ''
def test_reported_length():
data = "foo\n" * 499
m = MoreLimiter(10)
chunks = m.chunk(data)
assert chunks.initial_lines == 500 #plus the emptiness at the end.
assert chunks.next() == ("foo\n" * 10)
assert chunks.lines_left == 490
m.change_limit(5)
assert chunks.initial_lines == 500
assert chunks.next() == ("foo\n" * 5)
assert chunks.lines_left == 485