Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 1 | from unittest import TestCase |
| 2 | |
| 3 | import json.encoder |
Raymond Hettinger | c8d952d | 2009-05-27 06:50:31 +0000 | [diff] [blame] | 4 | from json import dumps |
| 5 | from collections import OrderedDict |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 6 | |
| 7 | CASES = [ |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 8 | ('/\\"\ucafe\ubabe\uab98\ufcde\ubcda\uef4a\x08\x0c\n\r\t`1~!@#$%^&*()_+-=[]{}|;:\',./<>?', '"/\\\\\\"\\ucafe\\ubabe\\uab98\\ufcde\\ubcda\\uef4a\\b\\f\\n\\r\\t`1~!@#$%^&*()_+-=[]{}|;:\',./<>?"'), |
| 9 | ('\u0123\u4567\u89ab\ucdef\uabcd\uef4a', '"\\u0123\\u4567\\u89ab\\ucdef\\uabcd\\uef4a"'), |
| 10 | ('controls', '"controls"'), |
| 11 | ('\x08\x0c\n\r\t', '"\\b\\f\\n\\r\\t"'), |
| 12 | ('{"object with 1 member":["array with 1 element"]}', '"{\\"object with 1 member\\":[\\"array with 1 element\\"]}"'), |
| 13 | (' s p a c e d ', '" s p a c e d "'), |
| 14 | ('\U0001d120', '"\\ud834\\udd20"'), |
| 15 | ('\u03b1\u03a9', '"\\u03b1\\u03a9"'), |
| 16 | ('\u03b1\u03a9', '"\\u03b1\\u03a9"'), |
| 17 | ('\u03b1\u03a9', '"\\u03b1\\u03a9"'), |
| 18 | ('\u03b1\u03a9', '"\\u03b1\\u03a9"'), |
| 19 | ("`1~!@#$%^&*()_+-={':[,]}|;.</>?", '"`1~!@#$%^&*()_+-={\':[,]}|;.</>?"'), |
| 20 | ('\x08\x0c\n\r\t', '"\\b\\f\\n\\r\\t"'), |
| 21 | ('\u0123\u4567\u89ab\ucdef\uabcd\uef4a', '"\\u0123\\u4567\\u89ab\\ucdef\\uabcd\\uef4a"'), |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 22 | ] |
| 23 | |
| 24 | class TestEncodeBaseStringAscii(TestCase): |
| 25 | def test_py_encode_basestring_ascii(self): |
| 26 | self._test_encode_basestring_ascii(json.encoder.py_encode_basestring_ascii) |
| 27 | |
| 28 | def test_c_encode_basestring_ascii(self): |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 29 | if not json.encoder.c_encode_basestring_ascii: |
| 30 | return |
| 31 | self._test_encode_basestring_ascii(json.encoder.c_encode_basestring_ascii) |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 32 | |
| 33 | def _test_encode_basestring_ascii(self, encode_basestring_ascii): |
| 34 | fname = encode_basestring_ascii.__name__ |
| 35 | for input_string, expect in CASES: |
| 36 | result = encode_basestring_ascii(input_string) |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 37 | self.assertEquals(result, expect, |
| 38 | '{0!r} != {1!r} for {2}({3!r})'.format( |
| 39 | result, expect, fname, input_string)) |
Raymond Hettinger | c8d952d | 2009-05-27 06:50:31 +0000 | [diff] [blame] | 40 | |
| 41 | def test_ordered_dict(self): |
| 42 | # See issue 6105 |
| 43 | items = [('one', 1), ('two', 2), ('three', 3), ('four', 4), ('five', 5)] |
| 44 | s = json.dumps(OrderedDict(items)) |
| 45 | self.assertEqual(s, '{"one": 1, "two": 2, "three": 3, "four": 4, "five": 5}') |
Raymond Hettinger | bcf6f92 | 2009-05-27 09:58:34 +0000 | [diff] [blame] | 46 | |
| 47 | def test_sorted_dict(self): |
| 48 | items = [('one', 1), ('two', 2), ('three', 3), ('four', 4), ('five', 5)] |
| 49 | s = json.dumps(dict(items), sort_keys=True) |
| 50 | self.assertEqual(s, '{"five": 5, "four": 4, "one": 1, "three": 3, "two": 2}') |