Rooba
/
tqdm
mirror of https://github.com/tqdm/tqdm.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Test that tqdm fails when iterating over a csv file

This commit is contained in:
Ivan Ivanov 2014-02-07 14:25:36 +13:00
parent 7b62451047
commit 4539ebc92d
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests.py
View File

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