mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-12 18:26:30 +03:00
* Fix matcher test
This commit is contained in:
parent
63e3d4e27f
commit
e89069dcae
|
@ -110,10 +110,7 @@ def test_overlap_prefix_reorder(EN):
|
||||||
def test_ner_interaction(EN):
|
def test_ner_interaction(EN):
|
||||||
EN.matcher.add('LAX_Airport', 'AIRPORT', {}, [[{ORTH: 'LAX'}]])
|
EN.matcher.add('LAX_Airport', 'AIRPORT', {}, [[{ORTH: 'LAX'}]])
|
||||||
EN.matcher.add('SFO_Airport', 'AIRPORT', {}, [[{ORTH: 'SFO'}]])
|
EN.matcher.add('SFO_Airport', 'AIRPORT', {}, [[{ORTH: 'SFO'}]])
|
||||||
doc = EN.tokenizer(u'get me a flight from SFO to LAX leaving 20 December and arriving on January 5th')
|
doc = EN(u'get me a flight from SFO to LAX leaving 20 December and arriving on January 5th')
|
||||||
EN.tagger(doc)
|
|
||||||
EN.matcher(doc)
|
|
||||||
EN.entity(doc)
|
|
||||||
|
|
||||||
ents = [(ent.label_, ent.text) for ent in doc.ents]
|
ents = [(ent.label_, ent.text) for ent in doc.ents]
|
||||||
assert ents[0] == ('AIRPORT', 'SFO')
|
assert ents[0] == ('AIRPORT', 'SFO')
|
||||||
|
|
Loading…
Reference in New Issue
Block a user