Add tests for all custom lex attrs

This commit is contained in:
Adriane Boyd 2023-05-08 11:11:03 +02:00
parent 01829933fd
commit 8fc4152297

View File

@ -0,0 +1,18 @@
import pytest
from spacy.lang.sr.lex_attrs import like_num, norm, prefix, suffix
@pytest.mark.parametrize(
"text,like_num,norm,prefix,suffix",
[
("нула", True, "nula", "n", "ula"),
("Казна", False, "kazna", "K", "zna"),
],
)
def test_lex_attrs(sr_tokenizer, text, like_num, norm, prefix, suffix):
tokens = sr_tokenizer(text)
assert len(tokens) == 1
assert tokens[0].like_num == like_num
assert tokens[0].norm_ == norm
assert tokens[0].prefix_ == prefix
assert tokens[0].suffix_ == suffix