blob: 9601bcb5035818ef9b127f56b09ab3ae429d9626 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
from fuzzycat.matching import anything_to_entity, match_release_fuzzy
from fatcat_openapi_client import ReleaseEntity
import pytest
import elasticsearch
import logging
logger = logging.getLogger('test_matching')
logger.setLevel(logging.DEBUG)
@pytest.fixture
def es_client():
return elasticsearch.Elasticsearch(["https://search.fatcat.wiki:443"])
@pytest.mark.skip(reason="we cannot use POST on es, which client uses: https://git.io/JLssk")
def test_match_release_fuzzy(es_client):
cases = (("wtv64ahbdzgwnan7rllwr3nurm", 2), )
for case, count in cases:
entity = anything_to_entity(case, ReleaseEntity)
logger.info(entity.title)
result = match_release_fuzzy(entity, es=es_client)
logger.info("given: {}".format(entity.title))
logger.info("found: {}".format(len(result)))
assert len(result) == count
|