Correct test

This commit is contained in:
richardpaulhudson 2023-01-12 21:08:33 +01:00
parent c763c500d8
commit c2c0a28ede

View File

@ -208,6 +208,16 @@ def test_incomplete_data(lowercasing: bool):
xp = lemmatizer.model.ops.xp
# Missing annotations.
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][:-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