Added CSV stream test

This commit is contained in:
Anthony Monthe
2019-06-27 23:19:06 +01:00
parent f55f56ae1d
commit 513bba2c20
2 changed files with 20 additions and 1 deletions
+1
View File
@@ -29,3 +29,4 @@ junit-py27.xml
# pyenv noise
.python-version
tablib.egg-info/*
+19 -1
View File
@@ -13,7 +13,7 @@ from uuid import uuid4
import tablib
from tablib.compat import markup, unicode, is_py3
from tablib.core import Row
from tablib.formats import csv as csv_format
from tablib.formats import _csv as csv_module
class TablibTestCase(unittest.TestCase):
@@ -262,6 +262,24 @@ class TablibTestCase(unittest.TestCase):
self.assertEqual(csv, self.founders.csv)
def test_csv_stream_export(self):
"""Verify exporting dataset object as CSV from file object."""
# Build up the csv string with headers first, followed by each row
csv = ''
for col in self.headers:
csv += col + ','
csv = csv.strip(',') + '\r\n'
for founder in self.founders:
for col in founder:
csv += str(col) + ','
csv = csv.strip(',') + '\r\n'
csv_stream = csv_module.export_stream_set(self.founders)
self.assertEqual(csv, csv_stream.getvalue())
def test_tsv_export(self):
"""Verify exporting dataset object as TSV."""