Add test for #1868: Vocab.__contains__ with ints

This commit is contained in:
Matthew Honnibal 2018-01-23 23:27:05 +01:00
parent 43f381ce36
commit 42a18ef903

View File

@ -0,0 +1,13 @@
'''Test Vocab.__contains__ works with int keys'''
from __future__ import unicode_literals
from ... vocab import Vocab
def test_issue1868():
vocab = Vocab()
lex = vocab['hello']
assert lex.orth in vocab
assert lex.orth_ in vocab
assert 'some string' not in vocab
int_id = vocab.strings.add('some string')
assert int_id not in vocab