From aaeb5c83600052f6696eb09a155112bc0252fffa Mon Sep 17 00:00:00 2001 From: Claude Paroz Date: Wed, 12 Feb 2020 20:12:53 +0100 Subject: Fixes #226 - Allow importing ragged CSV files (#456) --- tests/test_tablib.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'tests') diff --git a/tests/test_tablib.py b/tests/test_tablib.py index b20b669..3acdd44 100755 --- a/tests/test_tablib.py +++ b/tests/test_tablib.py @@ -804,6 +804,25 @@ class CSVTests(BaseTestCase): data.csv = csv_text self.assertEqual(data.width, 7) + def test_csv_import_set_ragged(self): + """Import CSV set when not all rows have the same length.""" + csv_text = ( + "H1,H2,H3\n" + "A,B\n" + "C,D,E\n" + "\n" + "F\n" + ) + dataset = tablib.import_set(csv_text, format="csv") + self.assertEqual( + str(dataset), + 'H1|H2|H3\n' + '--|--|--\n' + 'A |B | \n' + 'C |D |E \n' + 'F | | ' + ) + def test_csv_export(self): """Verify exporting dataset object as CSV.""" -- cgit v1.2.1