Fix test for word vector

This commit is contained in:
Matthew Honnibal 2016-12-18 23:31:55 +01:00
parent 13a0b31279
commit f38eb25fe1

View File

@ -8,11 +8,11 @@ import pytest
def test_vec(EN): def test_vec(EN):
hype = EN.vocab['hype'] hype = EN.vocab['hype']
assert hype.orth_ == 'hype' assert hype.orth_ == 'hype'
assert 0.08 >= hype.vector[0] > 0.07 assert -0.7 >= hype.vector[0] > -0.8
@pytest.mark.models @pytest.mark.models
def test_capitalized(EN): def test_capitalized(EN):
hype = EN.vocab['Hype'] hype = EN.vocab['Hype']
assert hype.orth_ == 'Hype' assert hype.orth_ == 'Hype'
assert 0.08 >= hype.vector[0] > 0.07 assert -0.7 >= hype.vector[0] > -0.8