mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-14 13:47:13 +03:00
68 lines
1.5 KiB
Python
68 lines
1.5 KiB
Python
"""Test that tokens are created correctly for whitespace."""
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
|
|
def test_single_space(en_tokenizer):
|
|
tokens = en_tokenizer('hello possums')
|
|
assert len(tokens) == 2
|
|
|
|
|
|
def test_double_space(en_tokenizer):
|
|
tokens = en_tokenizer('hello possums')
|
|
assert len(tokens) == 3
|
|
assert tokens[1].orth_ == ' '
|
|
|
|
|
|
def test_newline(en_tokenizer):
|
|
tokens = en_tokenizer('hello\npossums')
|
|
assert len(tokens) == 3
|
|
|
|
|
|
def test_newline_space(en_tokenizer):
|
|
tokens = en_tokenizer('hello \npossums')
|
|
assert len(tokens) == 3
|
|
|
|
|
|
def test_newline_double_space(en_tokenizer):
|
|
tokens = en_tokenizer('hello \npossums')
|
|
assert len(tokens) == 3
|
|
|
|
|
|
def test_newline_space_wrap(en_tokenizer):
|
|
tokens = en_tokenizer('hello \n possums')
|
|
assert len(tokens) == 3
|
|
|
|
|
|
def test_leading_space_offsets(en_tokenizer):
|
|
'''Issue #351
|
|
# this works
|
|
|
|
text1 = u"This is a cat."
|
|
a = english_spacy(text1)
|
|
|
|
tok0 = list(a.sents)[0][0]
|
|
print tok0, tok0.idx, text1[tok0.idx]
|
|
|
|
tok1 = list(a.sents)[0][1]
|
|
print tok1, tok1.idx, text1[tok1.idx]
|
|
|
|
print "=="
|
|
|
|
# this does not work
|
|
|
|
text2 = u" This is a cat."
|
|
b = english_spacy(text2)
|
|
|
|
tok0 = list(b.sents)[0][0]
|
|
print tok0, tok0.idx, text2[tok0.idx]
|
|
|
|
tok1 = list(b.sents)[0][1]
|
|
print tok1, tok1.idx, text2[tok1.idx]
|
|
'''
|
|
doc = en_tokenizer(u" This is a cat.")
|
|
assert doc[0].idx == 0
|
|
assert len(doc[0]) == 3
|
|
assert doc[1].idx == 3
|