spaCy/tests/test_vocab.py

from __future__ import unicode_literals
import pytest

from spacy.en import English


@pytest.fixture
def EN():
    return English()


def test_neq(EN):
    addr = EN.vocab['Hello']
    assert EN.vocab['bye'].sic != addr.sic


def test_eq(EN):
    addr = EN.vocab['Hello']
    assert EN.vocab['Hello'].sic == addr.sic


def test_case_neq(EN):
    addr = EN.vocab['Hello']
    assert EN.vocab['hello'].sic != addr.sic


def test_punct_neq(EN):
    addr = EN.vocab['Hello']
    assert EN.vocab['Hello,'].sic != addr.sic


def test_shape_attr(EN):
    example = EN.vocab['example']
    assert example.sic != example.shape