cpython/Lib/test/json_tests/test_indent.py

68 lines
1.8 KiB
Python
Raw Normal View History

import textwrap
from io import StringIO
2011-05-14 00:47:51 -03:00
from test.json_tests import PyTest, CTest
2011-05-14 00:47:51 -03:00
class TestIndent:
def test_indent(self):
h = [['blorpie'], ['whoops'], [], 'd-shtaeou', 'd-nthiouh', 'i-vhbjkhnth',
{'nifty': 87}, {'field': 'yes', 'morefield': False} ]
expect = textwrap.dedent("""\
[
\t[
\t\t"blorpie"
\t],
\t[
\t\t"whoops"
\t],
\t[],
\t"d-shtaeou",
\t"d-nthiouh",
\t"i-vhbjkhnth",
\t{
\t\t"nifty": 87
\t},
\t{
\t\t"field": "yes",
\t\t"morefield": false
\t}
]""")
2011-05-14 00:47:51 -03:00
d1 = self.dumps(h)
d2 = self.dumps(h, indent=2, sort_keys=True, separators=(',', ': '))
d3 = self.dumps(h, indent='\t', sort_keys=True, separators=(',', ': '))
d4 = self.dumps(h, indent=2, sort_keys=True)
d5 = self.dumps(h, indent='\t', sort_keys=True)
2011-05-14 00:47:51 -03:00
h1 = self.loads(d1)
h2 = self.loads(d2)
h3 = self.loads(d3)
self.assertEqual(h1, h)
self.assertEqual(h2, h)
self.assertEqual(h3, h)
self.assertEqual(d2, expect.expandtabs(2))
self.assertEqual(d3, expect)
self.assertEqual(d4, d2)
self.assertEqual(d5, d3)
def test_indent0(self):
h = {3: 1}
def check(indent, expected):
2011-05-14 00:47:51 -03:00
d1 = self.dumps(h, indent=indent)
self.assertEqual(d1, expected)
sio = StringIO()
2011-05-14 00:47:51 -03:00
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}')
2011-05-14 00:47:51 -03:00
class TestPyIndent(TestIndent, PyTest): pass
class TestCIndent(TestIndent, CTest): pass