summaryrefslogtreecommitdiff
path: root/test_tablib.py
diff options
context:
space:
mode:
Diffstat (limited to 'test_tablib.py')
-rwxr-xr-xtest_tablib.py35
1 files changed, 30 insertions, 5 deletions
diff --git a/test_tablib.py b/test_tablib.py
index 529b39d..3830dda 100755
--- a/test_tablib.py
+++ b/test_tablib.py
@@ -794,7 +794,7 @@ class TablibTestCase(unittest.TestCase):
self.assertEqual(third_row, expected_third)
- def test_unique(self):
+ def test_remove_duplicates(self):
"""Unique Rows."""
self.founders.append(self.john)
@@ -805,7 +805,7 @@ class TablibTestCase(unittest.TestCase):
self.assertEqual(self.founders[2], self.founders[5])
self.assertEqual(self.founders.height, 6)
- self.founders.unique()
+ self.founders.remove_duplicates()
self.assertEqual(self.founders[0], self.john)
self.assertEqual(self.founders[1], self.george)
@@ -832,16 +832,16 @@ class TablibTestCase(unittest.TestCase):
def test_subset(self):
"""Create a subset of a dataset"""
-
+
rows = (0, 2)
columns = ('first_name','gpa')
-
+
data.headers = self.headers
data.append(self.john)
data.append(self.george)
data.append(self.tom)
-
+
#Verify data is truncated
subset = data.subset(rows=rows, cols=columns)
self.assertEqual(type(subset), tablib.Dataset)
@@ -947,5 +947,30 @@ class TablibTestCase(unittest.TestCase):
except tablib.InvalidDatasetType:
self.fail("Subclass of tablib.Dataset should be accepted by Databook.add_sheet")
+
+ def test_csv_formatter_support_kwargs(self):
+ """Test CSV import and export with formatter configuration."""
+ data.append(self.john)
+ data.append(self.george)
+ data.headers = self.headers
+
+ expected = 'first_name;last_name;gpa\nJohn;Adams;90\nGeorge;Washington;67\n'
+
+ kwargs = dict(delimiter=';', lineterminator='\n')
+ _csv = data.export('csv', **kwargs)
+ self.assertEqual(expected, _csv)
+
+ # the import works but consider default delimiter=','
+ d1 = tablib.import_set(_csv, format="csv")
+ self.assertEqual(1, len(d1.headers))
+
+ d2 = tablib.import_set(_csv, format="csv", **kwargs)
+ self.assertEqual(3, len(d2.headers))
+
+ def test_databook_formatter_support_kwargs(self):
+ """Test XLSX export with formatter configuration."""
+ self.founders.export('xlsx', freeze_panes=False)
+
+
if __name__ == '__main__':
unittest.main()