mirror of https://github.com/explosion/spaCy.git
35 lines
936 B
Python
35 lines
936 B
Python
"""Test the Token.conjuncts property"""
|
|
from __future__ import unicode_literals
|
|
|
|
from spacy.en import English
|
|
import pytest
|
|
|
|
NLU = English()
|
|
|
|
def orths(tokens):
|
|
return [t.orth_ for t in tokens]
|
|
|
|
|
|
def test_simple_two():
|
|
tokens = NLU('I lost money and pride.')
|
|
pride = tokens[4]
|
|
assert orths(pride.conjuncts) == ['money', 'pride']
|
|
money = tokens[2]
|
|
assert orths(money.conjuncts) == ['money', 'pride']
|
|
|
|
|
|
def test_comma_three():
|
|
tokens = NLU('I found my wallet, phone and keys.')
|
|
keys = tokens[-2]
|
|
assert orths(keys.conjuncts) == ['wallet', 'phone', 'keys']
|
|
wallet = tokens[3]
|
|
assert orths(wallet.conjuncts) == ['wallet', 'phone', 'keys']
|
|
|
|
|
|
def test_and_three():
|
|
tokens = NLU('I found my wallet and phone and keys.')
|
|
keys = tokens[-2]
|
|
assert orths(keys.conjuncts) == ['wallet', 'phone', 'keys']
|
|
wallet = tokens[3]
|
|
assert orths(wallet.conjuncts) == ['wallet', 'phone', 'keys']
|