mirror of https://github.com/explosion/spaCy.git
* Add tests for string intern
This commit is contained in:
parent
7baef5b7ff
commit
168b2b8cb2
|
@ -0,0 +1,35 @@
|
||||||
|
# -*- 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):
|
||||||
|
with pytest.raises(TypeError):
|
||||||
|
A_i = sstore['A']
|
||||||
|
|
||||||
|
|
||||||
|
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]
|
Loading…
Reference in New Issue