def test_tsv_handler():
data = "a\tb\tc\n1\t2\t3\n4\t5\t6"
csv_file = cStringIO.StringIO(data)
handler = CSVHandler(csv_file)
test_run = handler.run()
result = test_run.next()
print result
assert result['a'] == '1'
assert result['b'] == '2'
assert result['c'] == '3'
result = test_run.next()
assert result['a'] == '4'
assert result['b'] == '5'
assert result['c'] == '6'
def test_csv_handler():
data = "a,b,c\n1,2,3\n4,5,6"
csv_file = cStringIO.StringIO(data)
handler = CSVHandler(csv_file)
test_run = handler.run()
result = test_run.next()
print result
assert result['a'] == '1'
assert result['b'] == '2'
assert result['c'] == '3'
result = test_run.next()
assert result['a'] == '4'
assert result['b'] == '5'
assert result['c'] == '6'