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

adjust test

Browse Source
This commit is contained in:
kadarakos 2023-05-08 13:21:37 +00:00
parent 530d812aed
commit 103654211f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/tests/pipeline/test_span_finder.py
View File

@ -63,7 +63,10 @@ def test_loss_alignment_example(tokens_predicted, tokens_reference, reference_tr
example.reference.spans[TRAINING_KEY] = [example.reference.char_span(5, 9)] example.reference.spans[TRAINING_KEY] = [example.reference.char_span(5, 9)]
span_finder = nlp.add_pipe("span_finder", config={"training_key": TRAINING_KEY}) span_finder = nlp.add_pipe("span_finder", config={"training_key": TRAINING_KEY})
nlp.initialize() nlp.initialize()
if predicted.text != reference.text:
with pytest.raises(ValueError, match="must match between reference and predicted"):
span_finder._get_aligned_truth_scores([example])
return
truth_scores = span_finder._get_aligned_truth_scores([example]) truth_scores = span_finder._get_aligned_truth_scores([example])
assert len(truth_scores) == len(tokens_predicted) assert len(truth_scores) == len(tokens_predicted)
assert truth_scores == reference_truths assert truth_scores == reference_truths