Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 1 | import pprint |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 2 | import test.test_support |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 3 | import unittest |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 4 | |
Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 5 | try: |
| 6 | uni = unicode |
| 7 | except NameError: |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 8 | def uni(x): |
| 9 | return x |
Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 10 | |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 11 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 12 | class QueryTestCase(unittest.TestCase): |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 13 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 14 | def setUp(self): |
| 15 | self.a = range(100) |
| 16 | self.b = range(200) |
| 17 | self.a[-12] = self.b |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 18 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 19 | def test_basic(self): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 20 | # Verify .isrecursive() and .isreadable() w/o recursion |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 21 | verify = self.assert_ |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 22 | pp = pprint.PrettyPrinter() |
Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 23 | for safe in (2, 2.0, 2j, "abc", [3], (2,2), {3: 3}, uni("yaddayadda"), |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 24 | self.a, self.b): |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 25 | # module-level convenience functions |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 26 | verify(not pprint.isrecursive(safe), |
| 27 | "expected not isrecursive for " + `safe`) |
| 28 | verify(pprint.isreadable(safe), |
| 29 | "expected isreadable for " + `safe`) |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 30 | # PrettyPrinter methods |
| 31 | verify(not pp.isrecursive(safe), |
| 32 | "expected not isrecursive for " + `safe`) |
| 33 | verify(pp.isreadable(safe), |
| 34 | "expected isreadable for " + `safe`) |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 35 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 36 | def test_knotted(self): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 37 | # Verify .isrecursive() and .isreadable() w/ recursion |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 38 | # Tie a knot. |
| 39 | self.b[67] = self.a |
| 40 | # Messy dict. |
| 41 | self.d = {} |
| 42 | self.d[0] = self.d[1] = self.d[2] = self.d |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 43 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 44 | verify = self.assert_ |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 45 | pp = pprint.PrettyPrinter() |
Tim Peters | a814db5 | 2001-05-14 07:05:58 +0000 | [diff] [blame] | 46 | |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 47 | for icky in self.a, self.b, self.d, (self.d, self.d): |
| 48 | verify(pprint.isrecursive(icky), "expected isrecursive") |
| 49 | verify(not pprint.isreadable(icky), "expected not isreadable") |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 50 | verify(pp.isrecursive(icky), "expected isrecursive") |
| 51 | verify(not pp.isreadable(icky), "expected not isreadable") |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 52 | |
| 53 | # Break the cycles. |
| 54 | self.d.clear() |
| 55 | del self.a[:] |
| 56 | del self.b[:] |
| 57 | |
| 58 | for safe in self.a, self.b, self.d, (self.d, self.d): |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 59 | # module-level convenience functions |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 60 | verify(not pprint.isrecursive(safe), |
| 61 | "expected not isrecursive for " + `safe`) |
| 62 | verify(pprint.isreadable(safe), |
| 63 | "expected isreadable for " + `safe`) |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 64 | # PrettyPrinter methods |
| 65 | verify(not pp.isrecursive(safe), |
| 66 | "expected not isrecursive for " + `safe`) |
| 67 | verify(pp.isreadable(safe), |
| 68 | "expected isreadable for " + `safe`) |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 69 | |
| 70 | def test_unreadable(self): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 71 | # Not recursive but not readable anyway |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 72 | verify = self.assert_ |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 73 | pp = pprint.PrettyPrinter() |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 74 | for unreadable in type(3), pprint, pprint.isrecursive: |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 75 | # module-level convenience functions |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 76 | verify(not pprint.isrecursive(unreadable), |
| 77 | "expected not isrecursive for " + `unreadable`) |
| 78 | verify(not pprint.isreadable(unreadable), |
| 79 | "expected not isreadable for " + `unreadable`) |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 80 | # PrettyPrinter methods |
| 81 | verify(not pp.isrecursive(unreadable), |
| 82 | "expected not isrecursive for " + `unreadable`) |
| 83 | verify(not pp.isreadable(unreadable), |
| 84 | "expected not isreadable for " + `unreadable`) |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 85 | |
Tim Peters | 95b3f78 | 2001-05-14 18:39:41 +0000 | [diff] [blame] | 86 | def test_same_as_repr(self): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 87 | # Simple objects and small containers that should be same as repr() |
Tim Peters | 95b3f78 | 2001-05-14 18:39:41 +0000 | [diff] [blame] | 88 | verify = self.assert_ |
Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 89 | for simple in (0, 0L, 0+0j, 0.0, "", uni(""), (), [], {}, verify, pprint, |
| 90 | -6, -6L, -6-6j, -1.5, "x", uni("x"), (3,), [3], {3: 6}, |
Tim Peters | 95b3f78 | 2001-05-14 18:39:41 +0000 | [diff] [blame] | 91 | (1,2), [3,4], {5: 6, 7: 8}, |
| 92 | {"xy\tab\n": (3,), 5: [[]], (): {}}, |
| 93 | range(10, -11, -1) |
| 94 | ): |
| 95 | native = repr(simple) |
| 96 | for function in "pformat", "saferepr": |
| 97 | f = getattr(pprint, function) |
| 98 | got = f(simple) |
| 99 | verify(native == got, "expected %s got %s from pprint.%s" % |
| 100 | (native, got, function)) |
Fred Drake | 43913dd | 2001-05-14 17:41:20 +0000 | [diff] [blame] | 101 | |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 102 | |
Barry Warsaw | 00859c0 | 2001-11-28 05:49:39 +0000 | [diff] [blame] | 103 | def test_basic_line_wrap(self): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 104 | # verify basic line-wrapping operation |
Barry Warsaw | 00859c0 | 2001-11-28 05:49:39 +0000 | [diff] [blame] | 105 | o = {'RPM_cal': 0, |
| 106 | 'RPM_cal2': 48059, |
| 107 | 'Speed_cal': 0, |
| 108 | 'controldesk_runtime_us': 0, |
| 109 | 'main_code_runtime_us': 0, |
| 110 | 'read_io_runtime_us': 0, |
| 111 | 'write_io_runtime_us': 43690} |
| 112 | exp = """\ |
| 113 | {'RPM_cal': 0, |
| 114 | 'RPM_cal2': 48059, |
| 115 | 'Speed_cal': 0, |
| 116 | 'controldesk_runtime_us': 0, |
| 117 | 'main_code_runtime_us': 0, |
| 118 | 'read_io_runtime_us': 0, |
| 119 | 'write_io_runtime_us': 43690}""" |
| 120 | self.assertEqual(pprint.pformat(o), exp) |
| 121 | |
Fred Drake | aee113d | 2002-04-02 05:08:35 +0000 | [diff] [blame] | 122 | def test_subclassing(self): |
| 123 | o = {'names with spaces': 'should be presented using repr()', |
| 124 | 'others.should.not.be': 'like.this'} |
| 125 | exp = """\ |
| 126 | {'names with spaces': 'should be presented using repr()', |
| 127 | others.should.not.be: like.this}""" |
| 128 | self.assertEqual(DottedPrettyPrinter().pformat(o), exp) |
| 129 | |
| 130 | |
| 131 | class DottedPrettyPrinter(pprint.PrettyPrinter): |
Guido van Rossum | 32c2ae7 | 2002-08-22 19:45:32 +0000 | [diff] [blame] | 132 | |
Fred Drake | aee113d | 2002-04-02 05:08:35 +0000 | [diff] [blame] | 133 | def format(self, object, context, maxlevels, level): |
| 134 | if isinstance(object, str): |
| 135 | if ' ' in object: |
| 136 | return `object`, 1, 0 |
| 137 | else: |
| 138 | return object, 0, 0 |
| 139 | else: |
| 140 | return pprint.PrettyPrinter.format( |
| 141 | self, object, context, maxlevels, level) |
| 142 | |
| 143 | |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 144 | def test_main(): |
Fred Drake | b456e4f | 2002-12-31 07:16:16 +0000 | [diff] [blame] | 145 | test.test_support.run_unittest(QueryTestCase) |
Fred Drake | 2e2be37 | 2001-09-20 21:33:42 +0000 | [diff] [blame] | 146 | |
| 147 | |
| 148 | if __name__ == "__main__": |
| 149 | test_main() |