32 lines
817 B
Python
32 lines
817 B
Python
import pytest
|
||
|
||
SQ_BASIC_TOKENIZATION_TESTS = [
|
||
(
|
||
"Askush nuk mund t’i nënshtrohet torturës ose dënimeve ose "
|
||
"trajtimeve çnjerëzore ose poshtëruese.",
|
||
[
|
||
"Askush",
|
||
"nuk",
|
||
"mund",
|
||
"t’i",
|
||
"nënshtrohet",
|
||
"torturës",
|
||
"ose",
|
||
"dënimeve",
|
||
"ose",
|
||
"trajtimeve",
|
||
"çnjerëzore",
|
||
"ose",
|
||
"poshtëruese",
|
||
".",
|
||
],
|
||
),
|
||
]
|
||
|
||
|
||
@pytest.mark.parametrize("text,expected_tokens", SQ_BASIC_TOKENIZATION_TESTS)
|
||
def test_sq_tokenizer_basic(sq_tokenizer, text, expected_tokens):
|
||
tokens = sq_tokenizer(text)
|
||
token_list = [token.text for token in tokens if not token.is_space]
|
||
assert expected_tokens == token_list
|