From 57a535f577379214f415a8deee5fa2ab2c5f2415 Mon Sep 17 00:00:00 2001 From: Hugo van Kemenade Date: Tue, 12 Nov 2019 10:53:20 +0200 Subject: Fix NameError: name '_get_column_widths' is not defined (#433) * Fix NameError: name '_get_column_widths' is not defined * Also test ReSTFormat.export_set --- tests/test_tablib.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'tests') diff --git a/tests/test_tablib.py b/tests/test_tablib.py index 61c1f3d..2d8798b 100755 --- a/tests/test_tablib.py +++ b/tests/test_tablib.py @@ -660,6 +660,28 @@ class RSTTests(BaseTestCase): '========== ========= ===' ) + def test_rst_export_set(self): + # Arrange + data = tablib.Dataset() + data.append(self.john) + data.headers = self.headers + fmt = registry.get_format("rst") + + # Act + out1 = fmt.export_set(data) + out2 = fmt.export_set_as_simple_table(data) + + # Assert + self.assertEqual(out1, out2) + self.assertEqual( + out1, + "========== ========= ===\n" + "first_name last_name gpa\n" + "========== ========= ===\n" + "John Adams 90 \n" + "========== ========= ===", + ) + class CSVTests(BaseTestCase): def test_csv_format_detect(self): -- cgit v1.2.1