16 lines
449 B
Python
16 lines
449 B
Python
import pytest
|
||
|
||
LG_BASIC_TOKENIZATION_TESTS = [
|
||
(
|
||
"Abooluganda ab’emmamba ababiri",
|
||
["Abooluganda", "ab’emmamba", "ababiri"],
|
||
),
|
||
]
|
||
|
||
|
||
@pytest.mark.parametrize("text,expected_tokens", LG_BASIC_TOKENIZATION_TESTS)
|
||
def test_lg_tokenizer_basic(lg_tokenizer, text, expected_tokens):
|
||
tokens = lg_tokenizer(text)
|
||
token_list = [token.text for token in tokens if not token.is_space]
|
||
assert expected_tokens == token_list
|