Test that tqdm fails when iterating over a csv file

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