zoo/tests/test_nanterre.py

28 lines
886 B
Python

import datetime
from zoo.zoo_nanterre.utils import PersonSearch
def test_person_search(db, rsu):
search = PersonSearch()
found = list(search.search_name(rsu[0].content['prenoms'], rsu[0].content['nom_de_naissance']))
assert rsu[0].id == found[0].id
assert found[0].similarity == 1.0
found = list(search.search_query(rsu[0].content['prenoms'] + ' ' +
rsu[0].content['nom_de_naissance']))
assert rsu[0].id == found[0].id
assert found[0].similarity == 1.0
birthdate = datetime.datetime.strptime(rsu[0].content['date_de_naissance'], '%Y-%m-%d').date()
found = list(search.search_birthdate(birthdate))
assert any(x for x in found if x.id == rsu[0].id)
assert len(found) == 1
found = list(search.search_email(rsu[0].content['email']))
assert found[0].id == rsu[0].id
assert len(found) == 1