From c2c0a28ede7d1f4d08418c9bf658d146dba042bc Mon Sep 17 00:00:00 2001 From: richardpaulhudson Date: Thu, 12 Jan 2023 21:08:33 +0100 Subject: [PATCH] Correct test --- .../pipeline/test_edit_tree_lemmatizer.py | 22 ++++++++++++++----- 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/spacy/tests/pipeline/test_edit_tree_lemmatizer.py b/spacy/tests/pipeline/test_edit_tree_lemmatizer.py index 18367ab38..1ad5fcacf 100644 --- a/spacy/tests/pipeline/test_edit_tree_lemmatizer.py +++ b/spacy/tests/pipeline/test_edit_tree_lemmatizer.py @@ -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])