| import textwrap | |
| from StringIO import StringIO | |
| from json.tests import PyTest, CTest | |
| class TestIndent(object): | |
| def test_indent(self): | |
| h = [['blorpie'], ['whoops'], [], 'd-shtaeou', 'd-nthiouh', 'i-vhbjkhnth', | |
| {'nifty': 87}, {'field': 'yes', 'morefield': False} ] | |
| expect = textwrap.dedent("""\ | |
| [ | |
| [ | |
| "blorpie" | |
| ], | |
| [ | |
| "whoops" | |
| ], | |
| [], | |
| "d-shtaeou", | |
| "d-nthiouh", | |
| "i-vhbjkhnth", | |
| { | |
| "nifty": 87 | |
| }, | |
| { | |
| "field": "yes", | |
| "morefield": false | |
| } | |
| ]""") | |
| d1 = self.dumps(h) | |
| d2 = self.dumps(h, indent=2, sort_keys=True, separators=(',', ': ')) | |
| h1 = self.loads(d1) | |
| h2 = self.loads(d2) | |
| self.assertEqual(h1, h) | |
| self.assertEqual(h2, h) | |
| self.assertEqual(d2, expect) | |
| def test_indent0(self): | |
| h = {3: 1} | |
| def check(indent, expected): | |
| d1 = self.dumps(h, indent=indent) | |
| self.assertEqual(d1, expected) | |
| sio = StringIO() | |
| self.json.dump(h, sio, indent=indent) | |
| self.assertEqual(sio.getvalue(), expected) | |
| # indent=0 should emit newlines | |
| check(0, '{\n"3": 1\n}') | |
| # indent=None is more compact | |
| check(None, '{"3": 1}') | |
| class TestPyIndent(TestIndent, PyTest): pass | |
| class TestCIndent(TestIndent, CTest): pass |