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']