Fix NameError: name '_get_column_widths' is not defined (#433)

* Fix NameError: name '_get_column_widths' is not defined

* Also test ReSTFormat.export_set

1 files changed, 22 insertions, 0 deletions

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):