mirror of https://github.com/explosion/spaCy.git
Add StringStore test for API docs
This commit is contained in:
parent
18b8050b07
commit
804dbb8d25
|
@ -6,6 +6,25 @@ from ...strings import StringStore
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
def test_stringstore_from_api_docs(stringstore):
|
||||||
|
apple_hash = stringstore.add('apple')
|
||||||
|
assert apple_hash == 8566208034543834098
|
||||||
|
assert stringstore[apple_hash] == u'apple'
|
||||||
|
|
||||||
|
assert u'apple' in stringstore
|
||||||
|
assert u'cherry' not in stringstore
|
||||||
|
|
||||||
|
orange_hash = stringstore.add('orange')
|
||||||
|
all_strings = [s for s in stringstore]
|
||||||
|
assert all_strings == [u'apple', u'orange']
|
||||||
|
|
||||||
|
banana_hash = stringstore.add('banana')
|
||||||
|
assert len(stringstore) == 3
|
||||||
|
assert banana_hash == 2525716904149915114
|
||||||
|
assert stringstore[banana_hash] == u'banana'
|
||||||
|
assert stringstore[u'banana'] == banana_hash
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('text1,text2,text3', [(b'Hello', b'goodbye', b'hello')])
|
@pytest.mark.parametrize('text1,text2,text3', [(b'Hello', b'goodbye', b'hello')])
|
||||||
def test_stringstore_save_bytes(stringstore, text1, text2, text3):
|
def test_stringstore_save_bytes(stringstore, text1, text2, text3):
|
||||||
key = stringstore.add(text1)
|
key = stringstore.add(text1)
|
||||||
|
|
Loading…
Reference in New Issue