def test_entity_type_to_json(self): class MyModel2(Entity): @classmethod def value_to_json(cls, value): return value + 1 entity_type = EntityType(MyModel2, 'title', 'description') self.assertEqual(entity_type.value_to_json(5), 6)
def test_entity_type_from_json(self): class MyModel3(Entity): @classmethod def json_to_value(cls, value): return value['value'] get_definitions()['MyModel3'] = { 'type': 'object', 'properties': { 'value': { 'type': 'integer' } } } entity_type = EntityType(MyModel3, 'title', 'description') self.assertEqual(entity_type.json_to_value({'value': 4}), 4)