def test_to_dict(self): entity_1 = Entity(mbid='f27ec8db-af05-4f36-916e-3d57f91ecf5e', type='test') db.session.add(entity_1) db.session.flush() self.assertDictEqual(entity_1.to_dict(), {'mbid': entity_1.mbid, 'type': entity_1.type, 'data': {}}) # Trying with data entity_2 = Entity(mbid='85d9c621-e30f-4788-a962-a089c0d34182', type='another_test', data='{"test": true}') db.session.add(entity_2) db.session.flush() self.assertDictEqual(entity_2.to_dict(), {'mbid': entity_2.mbid, 'type': entity_2.type, 'data': {'test': True}})