From aee113d368ebad8a51386baa7c0aec056aca94d5 Mon Sep 17 00:00:00 2001 From: Fred Drake Date: Tue, 2 Apr 2002 05:08:35 +0000 Subject: Add an experimental mechanism to support extending the pprint formatting. Partly responds to SF bug #505152. --- Lib/test/test_pprint.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'Lib/test/test_pprint.py') diff --git a/Lib/test/test_pprint.py b/Lib/test/test_pprint.py index 14626fb13f..d5e10aa458 100644 --- a/Lib/test/test_pprint.py +++ b/Lib/test/test_pprint.py @@ -96,6 +96,27 @@ class QueryTestCase(unittest.TestCase): 'write_io_runtime_us': 43690}""" self.assertEqual(pprint.pformat(o), exp) + def test_subclassing(self): + o = {'names with spaces': 'should be presented using repr()', + 'others.should.not.be': 'like.this'} + exp = """\ +{'names with spaces': 'should be presented using repr()', + others.should.not.be: like.this}""" + self.assertEqual(DottedPrettyPrinter().pformat(o), exp) + + +class DottedPrettyPrinter(pprint.PrettyPrinter): + def format(self, object, context, maxlevels, level): + if isinstance(object, str): + if ' ' in object: + return `object`, 1, 0 + else: + return object, 0, 0 + else: + return pprint.PrettyPrinter.format( + self, object, context, maxlevels, level) + + def test_main(): test_support.run_unittest(QueryTestCase) -- cgit v1.2.1