blob: 56999e6a357d2f43d7a67109f2051e8f2b1c4e64 (
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
27
28
29
|
from fuzzycat.matching import anything_to_entity, match_release_fuzzy
from fuzzycat.config import settings
from fatcat_openapi_client import ReleaseEntity
import pytest
import elasticsearch
import logging
logger = logging.getLogger('test_matching')
logger.setLevel(logging.DEBUG)
FATCAT_SEARCH_URL = settings.get("FATCAT_SEARCH_URL", "https://search.fatcat.wiki:443")
@pytest.fixture
def es_client():
return elasticsearch.Elasticsearch([FATCAT_SEARCH_URL])
@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
|
