spaCy/spacy/tests/th/test_tokenizer.py
Wannaphong Phatthiyaphaibun 1abf472068 add th test
2017-09-21 12:56:58 +07:00

14 lines
415 B
Python

# coding: utf-8
from __future__ import unicode_literals
import pytest
TOKENIZER_TESTS = [
("คุณรักผมไหม", ['คุณ', 'รัก', 'ผม', 'ไหม'])
]
@pytest.mark.parametrize('text,expected_tokens', TOKENIZER_TESTS)
def test_thai_tokenizer(th_tokenizer, text, expected_tokens):
tokens = [token.text for token in th_tokenizer(text)]
assert tokens == expected_tokens