def test_to_dict(self):
data = {
'id': 'some_uuid',
'extra_data': {
'test_foo': 5,
'fizz': 'buzz',
'spam': 'eggs'
},
'test_data': {
'bar': 'baz'
}
}
model = Model(data, 'test', {'fizz'})
serialized = {
'id': 'some_uuid',
'foo': 5,
'bar': 'baz',
'fizz': 'buzz',
'extra_data': {
'spam': 'eggs'
}
}
self.assertEqual(model.to_dict(), serialized)
serialized.pop('extra_data')
self.assertEqual(dict(model), serialized)
def test_to_dict(self):
data = {
'id': 'some_uuid',
'extra_data': {
'test_foo': 5,
'fizz': 'buzz',
'spam': 'eggs'
},
'test_data': {'bar': 'baz'}
}
model = Model(data, 'test', {'fizz'})
serialized = {
'id': 'some_uuid',
'foo': 5,
'bar': 'baz',
'fizz': 'buzz',
'extra_data': {'spam': 'eggs'}
}
self.assertEqual(model.to_dict(), serialized)
serialized.pop('extra_data')
self.assertEqual(dict(model), serialized)
