from __future__ import absolute_import import unittest import simplejson as json from simplejson.compat import StringIO try: from collections import namedtuple except ImportError: class Value(tuple): def __new__(cls, *args): return tuple.__new__(cls, args) def _asdict(self): return {'value': self[0]} class Point(tuple): def __new__(cls, *args): return tuple.__new__(cls, args) def _asdict(self): return {'x': self[0], 'y': self[1]} else: Value = namedtuple('Value', ['value']) Point = namedtuple('Point', ['x', 'y']) class DuckValue(object): def __init__(self, *args): self.value = Value(*args) def _asdict(self): return self.value._asdict() class DuckPoint(object): def __init__(self, *args): self.point = Point(*args) def _asdict(self): return self.point._asdict() class DeadDuck(object): _asdict = None class DeadDict(dict): _asdict = None CONSTRUCTORS = [ lambda v: v, lambda v: [v], lambda v: [{'key': v}], ] class TestNamedTuple(unittest.TestCase): def test_namedtuple_dumps(self): for v in [Value(1), Point(1, 2), DuckValue(1), DuckPoint(1, 2)]: d = v._asdict() self.assertEqual(d, json.loads(json.dumps(v))) self.assertEqual( d, json.loads(json.dumps(v, namedtuple_as_object=True))) self.assertEqual(d, json.loads(json.dumps(v, tuple_as_array=False))) self.assertEqual( d, json.loads(json.dumps(v, namedtuple_as_object=True, tuple_as_array=False))) def test_namedtuple_dumps_false(self): for v in [Value(1), Point(1, 2)]: l = list(v) self.assertEqual( l, json.loads(json.dumps(v, namedtuple_as_object=False))) self.assertRaises(TypeError, json.dumps, v, tuple_as_array=False, namedtuple_as_object=False) def test_namedtuple_dump(self): for v in [Value(1), Point(1, 2), DuckValue(1), DuckPoint(1, 2)]: d = v._asdict() sio = StringIO() json.dump(v, sio) self.assertEqual(d, json.loads(sio.getvalue())) sio = StringIO() json.dump(v, sio, namedtuple_as_object=True) self.assertEqual( d, json.loads(sio.getvalue())) sio = StringIO() json.dump(v, sio, tuple_as_array=False) self.assertEqual(d, json.loads(sio.getvalue())) sio = StringIO() json.dump(v, sio, namedtuple_as_object=True, tuple_as_array=False) self.assertEqual( d, json.loads(sio.getvalue())) def test_namedtuple_dump_false(self): for v in [Value(1), Point(1, 2)]: l = list(v) sio = StringIO() json.dump(v, sio, namedtuple_as_object=False) self.assertEqual( l, json.loads(sio.getvalue())) self.assertRaises(TypeError, json.dump, v, StringIO(), tuple_as_array=False, namedtuple_as_object=False) def test_asdict_not_callable_dump(self): for f in CONSTRUCTORS: self.assertRaises(TypeError, json.dump, f(DeadDuck()), StringIO(), namedtuple_as_object=True) sio = StringIO() json.dump(f(DeadDict()), sio, namedtuple_as_object=True) self.assertEqual( json.dumps(f({})), sio.getvalue()) def test_asdict_not_callable_dumps(self): for f in CONSTRUCTORS: self.assertRaises(TypeError, json.dumps, f(DeadDuck()), namedtuple_as_object=True) self.assertEqual( json.dumps(f({})), json.dumps(f(DeadDict()), namedtuple_as_object=True))
Name | Type | Size | Permission | Actions |
---|---|---|---|---|
__pycache__ | Folder | 0755 |
|
|
__init__.py | File | 2.1 KB | 0644 |
|
test_bigint_as_string.py | File | 2.19 KB | 0644 |
|
test_bitsize_int_as_string.py | File | 2.24 KB | 0644 |
|
test_check_circular.py | File | 917 B | 0644 |
|
test_decimal.py | File | 2.48 KB | 0644 |
|
test_decode.py | File | 4.72 KB | 0644 |
|
test_default.py | File | 221 B | 0644 |
|
test_dump.py | File | 10.11 KB | 0644 |
|
test_encode_basestring_ascii.py | File | 2.28 KB | 0644 |
|
test_encode_for_html.py | File | 1.48 KB | 0644 |
|
test_errors.py | File | 2.03 KB | 0644 |
|
test_fail.py | File | 6.2 KB | 0644 |
|
test_float.py | File | 1.4 KB | 0644 |
|
test_for_json.py | File | 2.7 KB | 0644 |
|
test_indent.py | File | 2.51 KB | 0644 |
|
test_item_sort_key.py | File | 1.34 KB | 0644 |
|
test_iterable.py | File | 1.36 KB | 0644 |
|
test_namedtuple.py | File | 3.91 KB | 0644 |
|
test_pass1.py | File | 1.71 KB | 0644 |
|
test_pass2.py | File | 386 B | 0644 |
|
test_pass3.py | File | 482 B | 0644 |
|
test_raw_json.py | File | 1.04 KB | 0644 |
|
test_recursion.py | File | 1.64 KB | 0644 |
|
test_scanstring.py | File | 7.22 KB | 0644 |
|
test_separators.py | File | 942 B | 0644 |
|
test_speedups.py | File | 4.05 KB | 0644 |
|
test_str_subclass.py | File | 740 B | 0644 |
|
test_subclass.py | File | 1.1 KB | 0644 |
|
test_tool.py | File | 3.23 KB | 0644 |
|
test_tuple.py | File | 1.79 KB | 0644 |
|
test_unicode.py | File | 6.89 KB | 0644 |
|