spaCy/spacy/tests/vocab/test_lexeme.py

43 lines
839 B
Python

from __future__ import unicode_literals
import pytest
from spacy.attrs import *
def test_lexeme_eq(en_vocab):
'''Test Issue #361: Equality of lexemes'''
cat1 = en_vocab['cat']
cat2 = en_vocab['cat']
assert cat1 == cat2
def test_lexeme_neq(en_vocab):
'''Inequality of lexemes'''
cat = en_vocab['cat']
dog = en_vocab['dog']
assert cat != dog
def test_lexeme_lt(en_vocab):
'''More frequent is l.t. less frequent'''
noun = en_vocab['NOUN']
opera = en_vocab['opera']
assert noun < opera
assert opera > noun
def test_lexeme_hash(en_vocab):
'''Test that lexemes are hashable.'''
phantom = en_vocab['phantom']
opera = en_vocab['opera']
lexes = {phantom: phantom, opera: opera}
assert lexes[phantom].orth_ == 'phantom'
assert lexes[opera].orth_ == 'opera'