diff options
Diffstat (limited to 'test_tablib.py')
| -rwxr-xr-x | test_tablib.py | 35 |
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() |
