mirror of https://github.com/explosion/spaCy.git
40 lines
870 B
Python
40 lines
870 B
Python
# -*- coding: utf8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from spacy.utf8string import StringStore
|
|
|
|
import pytest
|
|
|
|
@pytest.fixture
|
|
def sstore():
|
|
return StringStore()
|
|
|
|
def test_save_bytes(sstore):
|
|
Hello_i = sstore[b'Hello']
|
|
assert Hello_i == 1
|
|
assert sstore[b'Hello'] == 1
|
|
assert sstore[b'goodbye'] != Hello_i
|
|
assert sstore[b'hello'] != Hello_i
|
|
assert Hello_i == 1
|
|
|
|
|
|
def test_save_unicode(sstore):
|
|
Hello_i = sstore[u'Hello']
|
|
assert Hello_i == 1
|
|
assert sstore[u'Hello'] == 1
|
|
assert sstore[u'goodbye'] != Hello_i
|
|
assert sstore[u'hello'] != Hello_i
|
|
assert Hello_i == 1
|
|
|
|
|
|
def test_zero_id(sstore):
|
|
with pytest.raises(IndexError):
|
|
sstore[0]
|
|
|
|
def test_retrieve_id(sstore):
|
|
A_i = sstore[b'A']
|
|
assert sstore.size == 1
|
|
assert sstore[1] == b'A'
|
|
with pytest.raises(IndexError):
|
|
sstore[2]
|