From 8fc4152297a2bbc14003e0132da4e7764a609d9a Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Mon, 8 May 2023 11:11:03 +0200 Subject: [PATCH] Add tests for all custom lex attrs --- spacy/tests/lang/sr/test_lex_attrs.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 spacy/tests/lang/sr/test_lex_attrs.py diff --git a/spacy/tests/lang/sr/test_lex_attrs.py b/spacy/tests/lang/sr/test_lex_attrs.py new file mode 100644 index 000000000..54da62c35 --- /dev/null +++ b/spacy/tests/lang/sr/test_lex_attrs.py @@ -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