add test for get_markdown and get_rst

1 files changed, 32 insertions, 0 deletions

diff --git a/tests/test_prettytable.py b/tests/test_prettytable.py
index 6d8c573..235cd81 100644
--- a/tests/test_prettytable.py
+++ b/tests/test_prettytable.py
@@ -727,5 +727,37 @@ g..
 """.strip())
 
 
+class PrintMarkdownAndRstTest(unittest.TestCase):
+    def setUp(self):
+        self.x = PrettyTable(["A", "B", "C"])
+        self.x.add_row(["a", "b", "c"])
+        self.x.add_row(["aa", "bb", "cc"])
+
+    def testMarkdownOutput(self):
+        result = self.x.get_markdown()
+        print()
+        print(result)
+        self.assertEqual(result.strip(), """
+| A  | B  | C  |
+|----|----|----|
+| a  | b  | c  |
+| aa | bb | cc |
+""".strip())
+
+    def testRstOutput(self):
+        result = self.x.get_rst()
+        print()
+        print(result)
+        self.assertEqual(result.strip(), """
++----+----+----+
+| A  | B  | C  |
++====+====+====+
+| a  | b  | c  |
++----+----+----+
+| aa | bb | cc |
++----+----+----+
+""".strip())
+
+
 if __name__ == "__main__":
     unittest.main()