Merge pull request #377 from maxbachmann-forks/master

use rapidfuzz instead of fuzzywuzzy
pull/1/head
Chris Lenk 2020-07-08 10:49:41 -04:00 committed by GitHub
commit 457085a115
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 5 additions and 6 deletions

View File

@ -1,7 +1,7 @@
os: linux
language: python
cache: pip
dist: xenial
dist: bionic
python:
- "3.5"
- "3.6"

View File

@ -61,6 +61,6 @@ setup(
},
extras_require={
'taxii': ['taxii2-client'],
'semantic': ['haversine', 'fuzzywuzzy'],
'semantic': ['haversine', 'rapidfuzz'],
},
)

View File

@ -364,8 +364,8 @@ def partial_string_based(str1, str2):
float: Number between 0.0 and 1.0 depending on match criteria.
"""
from fuzzywuzzy import fuzz
result = fuzz.token_sort_ratio(str1, str2, force_ascii=False)
from rapidfuzz import fuzz
result = round(fuzz.token_sort_ratio(str1, str2))
logger.debug("--\t\tpartial_string_based '%s' '%s'\tresult: '%s'", str1, str2, result)
return result / 100.0

View File

@ -9,9 +9,8 @@ deps =
pytest-cov
coverage
taxii2-client
fuzzywuzzy
rapidfuzz
haversine
python-Levenshtein
medallion
commands =
python -m pytest --cov=stix2 stix2/test/ --cov-report term-missing -W ignore::stix2.exceptions.STIXDeprecationWarning