diff --git a/tests.py b/tests.py index f003757a..8f515ef8 100644 --- a/tests.py +++ b/tests.py @@ -1,5 +1,7 @@ from __future__ import unicode_literals -from tqdm import format_interval, format_meter +from StringIO import StringIO +import csv +from tqdm import format_interval, format_meter, tqdm def test_format_interval(): @@ -15,3 +17,17 @@ def test_format_meter(): assert format_meter(231, 1000, 392) == \ "|##--------| 231/1000 23% [elapsed: " \ "06:32 left: 21:44, 0.59 iters/sec]" + + +def test_iterate_over_csv_rows(): + # Create a test csv pseudo file + test_csv_file = StringIO() + writer = csv.writer(test_csv_file) + for i in range(3): + writer.writerow(['test', 'test', 'test']) + test_csv_file.seek(0) + + # Test that nothing fails if we iterate over rows + reader = csv.DictReader(test_csv_file, fieldnames=('row1', 'row2', 'row3')) + for row in tqdm(reader): + pass