spaCy/tests/test_contractions.py

42 lines
993 B
Python

from __future__ import unicode_literals
from spacy.en import tokenize, lookup, unhash
def test_possess():
tokens = tokenize("Mike's")
assert unhash(tokens[0].lex) == "Mike"
assert unhash(tokens[1].lex) == "'s"
assert len(tokens) == 2
def test_apostrophe():
tokens = tokenize("schools'")
assert len(tokens) == 2
assert unhash(tokens[1].lex) == "'"
assert unhash(tokens[0].lex) == "schools"
def test_LL():
tokens = tokenize("we'll")
assert len(tokens) == 2
assert unhash(tokens[1].lex) == "will"
assert unhash(tokens[0].lex) == "we"
def test_aint():
tokens = tokenize("ain't")
assert len(tokens) == 2
assert unhash(tokens[0].lex) == "are"
assert unhash(tokens[1].lex) == "not"
def test_capitalized():
tokens = tokenize("can't")
assert len(tokens) == 2
tokens = tokenize("Can't")
assert len(tokens) == 2
tokens = tokenize("Ain't")
assert len(tokens) == 2
assert unhash(tokens[0].lex) == "Are"