Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

* Upd tests

This commit is contained in:
Matthew Honnibal 2015-09-12 04:23:35 +02:00
parent c808cd50d9
commit 6d38c3cff0
1 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/spans/test_times.py
View File

@ -2,20 +2,21 @@ from __future__ import unicode_literals
import pytest import pytest
# This approach is deprecated for now
@pytest.mark.models #
def test_am_pm(en_nlp): #@pytest.mark.models
numbers = ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12'] #def test_am_pm(en_nlp):
variants = ['a.m.', 'am', 'p.m.', 'pm'] # numbers = ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12']
spaces = ['', ' '] # variants = ['a.m.', 'am', 'p.m.', 'pm']
for num in numbers: # spaces = ['', ' ']
for var in variants: # for num in numbers:
for space in spaces: # for var in variants:
string = u"The meeting was at %s%s%s wasn't it?" % (num, space, var) # for space in spaces:
tokens = en_nlp(string, merge_mwes=True) # string = u"The meeting was at %s%s%s wasn't it?" % (num, space, var)
assert tokens[4].orth_ == '%s%s%s' % (num, space, var) # tokens = en_nlp(string, merge_mwes=True)
ents = list(tokens.ents) # assert tokens[4].orth_ == '%s%s%s' % (num, space, var)
assert len(ents) == 1, ents # ents = list(tokens.ents)
assert ents[0].label_ == 'TIME', string # assert len(ents) == 1, ents
if ents[0].start == 4 and ents[0].end == 5: # assert ents[0].label_ == 'TIME', string
assert ents[0].orth_ == '%s%s%s' % (num, space, var) # if ents[0].start == 4 and ents[0].end == 5:
# assert ents[0].orth_ == '%s%s%s' % (num, space, var)