explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-24 14:04:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Correct test

Browse Source
This commit is contained in:
richardpaulhudson 2023-01-12 21:08:33 +01:00
parent c763c500d8
commit c2c0a28ede
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
spacy/tests/pipeline/test_edit_tree_lemmatizer.py
View File

@ -208,13 +208,23 @@ def test_incomplete_data(lowercasing: bool):
xp = lemmatizer.model.ops.xp
# Missing annotations.
assert xp.count_nonzero(dX[0][0]) == 0
assert xp.count_nonzero(dX[0][3]) == 0
assert xp.count_nonzero(dX[1][0]) == 0
assert xp.count_nonzero(dX[1][3]) == 0
if lowercasing:
assert xp.count_nonzero(dX[0][0][:-1]) == 0
assert xp.count_nonzero(dX[0][3][:-1]) == 0
assert xp.count_nonzero(dX[1][0][:-1]) == 0
assert xp.count_nonzero(dX[1][3][:-1]) == 0
# Misaligned annotations.
assert xp.count_nonzero(dX[1][1]) == 0
# Misaligned annotations.
assert xp.count_nonzero(dX[1][1][:-1]) == 0
else:
assert xp.count_nonzero(dX[0][0]) == 0
assert xp.count_nonzero(dX[0][3]) == 0
assert xp.count_nonzero(dX[1][0]) == 0
assert xp.count_nonzero(dX[1][3]) == 0
# Misaligned annotations.
assert xp.count_nonzero(dX[1][1]) == 0
@pytest.mark.parametrize("lowercasing", [True, False])