spaCy/tests/test_post_punct.py
2014-11-02 21:24:09 +11:00

48 lines
1.1 KiB
Python

from __future__ import unicode_literals
from spacy.en import EN
import pytest
@pytest.fixture
def close_puncts():
return [')', ']', '}', '*']
def test_close(close_puncts):
word_str = 'Hello'
for p in close_puncts:
string = word_str + p
tokens = EN.tokenize(string)
assert len(tokens) == 2
assert tokens[1].string == p
assert tokens[0].string == word_str
def test_two_different_close(close_puncts):
word_str = 'Hello'
for p in close_puncts:
string = word_str + p + "'"
tokens = EN.tokenize(string)
assert len(tokens) == 3
assert tokens[0].string == word_str
assert tokens[1].string == p
assert tokens[2].string == "'"
def test_three_same_close(close_puncts):
word_str = 'Hello'
for p in close_puncts:
string = word_str + p + p + p
tokens = EN.tokenize(string)
assert len(tokens) == 4
assert tokens[0].string == word_str
assert tokens[1].string == p
def test_double_end_quote():
assert len(EN.tokenize("Hello''")) == 2
assert len(EN.tokenize("''")) == 1