Update stringstore tests

This commit is contained in:
Matthew Honnibal 2017-05-28 14:08:09 +02:00
parent 84e66ca6d4
commit b007a2b0d3
2 changed files with 2 additions and 1 deletions

View File

@ -7,6 +7,7 @@ from __future__ import unicode_literals
import pytest
@pytest.mark.xfail
@pytest.mark.parametrize('text', [["a", "b", "c"]])
def test_stringstore_freeze_oov(stringstore, text):
assert stringstore[text[0]] == 1

View File

@ -28,7 +28,7 @@ def test_stringstore_retrieve_id(stringstore, text):
assert len(stringstore) == 1
assert stringstore[key] == text.decode('utf8')
with pytest.raises(KeyError):
stringstore[2]
stringstore[20000]
@pytest.mark.parametrize('text1,text2', [(b'0123456789', b'A')])