mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-21 11:29:13 +03:00
28 lines
745 B
Python
28 lines
745 B
Python
# coding: utf-8
|
|
"""Test that tokenizer exceptions are handled correctly."""
|
|
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize('text', ["e.g.", "p.m.", "Jan.", "Dec.", "Inc."])
|
|
def test_tokenizer_handles_abbr(en_tokenizer, text):
|
|
tokens = en_tokenizer(text)
|
|
assert len(tokens) == 1
|
|
|
|
|
|
def test_tokenizer_handles_exc_in_text(en_tokenizer):
|
|
text = "It's mediocre i.e. bad."
|
|
tokens = en_tokenizer(text)
|
|
assert len(tokens) == 6
|
|
assert tokens[3].text == "i.e."
|
|
|
|
|
|
@pytest.mark.parametrize('text', ["1am", "12a.m.", "11p.m.", "4pm"])
|
|
def test_tokenizer_handles_times(en_tokenizer, text):
|
|
tokens = en_tokenizer(text)
|
|
assert len(tokens) == 2
|
|
assert tokens[1].lemma_ in ["a.m.", "p.m."]
|