* Add tests for word shape

This commit is contained in:
Matthew Honnibal 2014-08-30 19:00:10 +02:00
parent 4e5b2d47e2
commit 6209d94f83
1 changed files with 27 additions and 0 deletions

27
tests/test_shape.py Normal file
View File

@ -0,0 +1,27 @@
from __future__ import unicode_literals
import pytest
from spacy.orth import word_shape as ws
def test_capitalized():
assert ws('Nasa') == 'Xxxx'
def test_truncate():
assert ws('capitalized') == 'xxx'
def test_digits():
assert ws('999999999') == 'ddd'
def test_mix():
assert ws('C3P0') == 'XdXd'
def test_punct():
assert ws(',') == ','
def test_space():
assert ws('\n') == '\n'
def test_punct_seq():
assert ws('``,-') == '``,-'