test iter_lines

This commit is contained in:
Kenneth Reitz
2011-12-14 10:34:31 -05:00
parent 47510486ec
commit 5a3aae73ba
+11
View File
@@ -592,5 +592,16 @@ class RequestsTestSuite(unittest.TestCase):
assert r2._content
assert r2.content
def test_iter_lines(self):
lines = (0, 2, 10, 100)
for i in lines:
r = requests.get(httpbin('stream', str(i)), prefetch=False)
len_lines = len([l for l in r.iter_lines()])
self.assertEqual(i, len_lines)
if __name__ == '__main__':
unittest.main()