explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-10 00:20:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

update tests from master to follow v4 principles

Browse Source
This commit is contained in:
svlandeg 2023-01-11 18:57:50 +01:00
parent b2fd9490e3
commit 6ff5eb256c
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
spacy/tests/pipeline/test_entity_ruler.py
View File

@ -353,9 +353,9 @@ def test_entity_ruler_overlapping_spans(nlp):
assert doc.ents[0].label_ == "FOOBAR" assert doc.ents[0].label_ == "FOOBAR"
@pytest.mark.parametrize("entity_ruler_factory", ENTITY_RULERS) @pytest.mark.parametrize()
def test_entity_ruler_fuzzy_pipe(nlp, entity_ruler_factory): def test_entity_ruler_fuzzy_pipe(nlp):
ruler = nlp.add_pipe(entity_ruler_factory, name="entity_ruler") ruler = nlp.add_pipe("entity_ruler")
patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}] patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}]
ruler.add_patterns(patterns) ruler.add_patterns(patterns)
doc = nlp("helloo") doc = nlp("helloo")
@ -363,9 +363,9 @@ def test_entity_ruler_fuzzy_pipe(nlp, entity_ruler_factory):
assert doc.ents[0].label_ == "HELLO" assert doc.ents[0].label_ == "HELLO"
@pytest.mark.parametrize("entity_ruler_factory", ENTITY_RULERS) @pytest.mark.parametrize()
def test_entity_ruler_fuzzy(nlp, entity_ruler_factory): def test_entity_ruler_fuzzy(nlp):
ruler = nlp.add_pipe(entity_ruler_factory, name="entity_ruler") ruler = nlp.add_pipe("entity_ruler")
patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}] patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}]
ruler.add_patterns(patterns) ruler.add_patterns(patterns)
doc = nlp("helloo") doc = nlp("helloo")
@ -373,15 +373,14 @@ def test_entity_ruler_fuzzy(nlp, entity_ruler_factory):
assert doc.ents[0].label_ == "HELLO" assert doc.ents[0].label_ == "HELLO"
@pytest.mark.parametrize("entity_ruler_factory", ENTITY_RULERS) @pytest.mark.parametrize()
def test_entity_ruler_fuzzy_disabled(nlp, entity_ruler_factory): def test_entity_ruler_fuzzy_disabled(nlp):
@registry.misc("test_fuzzy_compare_disabled") @registry.misc("test_fuzzy_compare_disabled")
def make_test_fuzzy_compare_disabled(): def make_test_fuzzy_compare_disabled():
return lambda x, y, z: False return lambda x, y, z: False
ruler = nlp.add_pipe( ruler = nlp.add_pipe(
entity_ruler_factory, "entity_ruler",
name="entity_ruler",
config={"matcher_fuzzy_compare": {"@misc": "test_fuzzy_compare_disabled"}}, config={"matcher_fuzzy_compare": {"@misc": "test_fuzzy_compare_disabled"}},
) )
patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}] patterns = [{"label": "HELLO", "pattern": [{"LOWER": {"FUZZY": "hello"}}]}]