explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-25 11:23:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use a single variable for tests

Browse Source
This commit is contained in:
Lj Miranda 2022-11-29 11:25:35 +08:00
parent ac0ac3eb99
commit 4ab27d4517
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
spacy/tests/pipeline/test_spancat.py
View File

@ -15,6 +15,8 @@ OPS = get_current_ops()
SPAN_KEY = "labeled_spans" SPAN_KEY = "labeled_spans"
SPANCAT_COMPONENTS = ["spancat", "spancat_exclusive"]
TRAIN_DATA = [ TRAIN_DATA = [
("Who is Shaka Khan?", {"spans": {SPAN_KEY: [(7, 17, "PERSON")]}}), ("Who is Shaka Khan?", {"spans": {SPAN_KEY: [(7, 17, "PERSON")]}}),
( (
@ -41,7 +43,7 @@ def make_examples(nlp, data=TRAIN_DATA):
return train_examples return train_examples
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_no_label(name): def test_no_label(name):
nlp = Language() nlp = Language()
nlp.add_pipe(name, config={"spans_key": SPAN_KEY}) nlp.add_pipe(name, config={"spans_key": SPAN_KEY})
@ -49,7 +51,7 @@ def test_no_label(name):
nlp.initialize() nlp.initialize()
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_no_resize(name): def test_no_resize(name):
nlp = Language() nlp = Language()
spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY}) spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY})
@ -63,7 +65,7 @@ def test_no_resize(name):
spancat.add_label("Stuff") spancat.add_label("Stuff")
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_implicit_labels(name): def test_implicit_labels(name):
nlp = Language() nlp = Language()
spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY}) spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY})
@ -73,7 +75,7 @@ def test_implicit_labels(name):
assert spancat.labels == ("PERSON", "LOC") assert spancat.labels == ("PERSON", "LOC")
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_explicit_labels(name): def test_explicit_labels(name):
nlp = Language() nlp = Language()
spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY}) spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY})
@ -375,7 +377,7 @@ def test_overfitting_IO_overlapping():
assert set([span.label_ for span in spans2]) == {"LOC", "DOUBLE_LOC"} assert set([span.label_ for span in spans2]) == {"LOC", "DOUBLE_LOC"}
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_zero_suggestions(name): def test_zero_suggestions(name):
# Test with a suggester that returns 0 suggestions # Test with a suggester that returns 0 suggestions
@ -404,7 +406,7 @@ def test_zero_suggestions(name):
nlp.update(train_examples, sgd=optimizer) nlp.update(train_examples, sgd=optimizer)
@pytest.mark.parametrize("name", ["spancat", "spancat_exclusive"]) @pytest.mark.parametrize("name", SPANCAT_COMPONENTS)
def test_set_candidates(name): def test_set_candidates(name):
nlp = Language() nlp = Language()
spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY}) spancat = nlp.add_pipe(name, config={"spans_key": SPAN_KEY})