From aee113d368ebad8a51386baa7c0aec056aca94d5 Mon Sep 17 00:00:00 2001
From: Fred Drake <fdrake@acm.org>
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