diff options
Diffstat (limited to 'tests/test_prettytable.py')
-rw-r--r-- | tests/test_prettytable.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tests/test_prettytable.py b/tests/test_prettytable.py index 6d8c573..064e2b0 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_md_string() + 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_string() + print() + print(result) + self.assertEqual(result.strip(), """ ++----+----+----+ +| A | B | C | ++====+====+====+ +| a | b | c | ++----+----+----+ +| aa | bb | cc | ++----+----+----+ +""".strip()) + + if __name__ == "__main__": unittest.main() |