def test_normalize_lemmatize_fields():
exist_field = normalize.lemmatize_fields[0]
exist_data = 'коты'
data = {exist_field: exist_data}
normalize.normalize(data)
for field in normalize.lemmatize_fields:
if field == exist_field:
assert data[field] == normalize.lemmatize(exist_data)
else:
assert data[field] == []
def save(self, data):
normalize(data)
self.connection[DB.DB_NAME][self.user_id]\
.replace_one({'id': data['id']}, data, upsert=True)
def test_normalize_occupation():
data = {'occupation': {"type": "university", "id": 227, "name": "МГППУ"}}
normalize.normalize(data)
assert data['occupation'] == 'МГППУ'
def test_normalize_personal():
data = {'personal': {'personal_field': 'personal_dta', 'some': 'thing'}}
normalize.normalize(data)
assert 'personal' in data
assert data['personal_field'] == 'personal_dta'
assert data['some'] == 'thing'
def test_normalize_array_fields():
data = {'schools': [{'id': 5}, {'id': 7}]}
normalize.normalize(data)
assert data['schools'] == [5, 7]
assert data['universities'] == []
def test_normalize_obj_fields(data, expected, field):
normalize.normalize(data)
assert data[field] == expected
def load_settings(self, vk):
if self.load_from_file():
return
user = vk.get_user(self.user_id)
normalize.normalize(user)
self.load_from_vk(user)