spaCy/tests/test_vec.py

from __future__ import unicode_literals

from spacy.en import English

import pytest


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


def test_vec(EN):
    hype = EN.vocab['hype']
    assert hype.orth_ == 'hype'
    assert 0.08 >= hype.repvec[0] > 0.07


def test_capitalized(EN):
    hype = EN.vocab['Hype']
    assert hype.orth_ == 'Hype'
    assert 0.08 >= hype.repvec[0] > 0.07