spaCy/spacy/tests/regression/test_issue1375.py

from __future__ import unicode_literals
import pytest
from ...vocab import Vocab
from ...tokens.doc import Doc


def test_issue1375():
    '''Test that token.nbor() raises IndexError for out-of-bounds access.'''
    doc = Doc(Vocab(), words=['0', '1', '2'])
    with pytest.raises(IndexError):
        assert doc[0].nbor(-1)
    assert doc[1].nbor(-1).text == '0'
    with pytest.raises(IndexError):
        assert doc[2].nbor(1)
    assert doc[1].nbor(1).text == '2'