mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-13 13:17:06 +03:00
25 lines
577 B
Python
25 lines
577 B
Python
|
from ...attrs import LOWER, ORTH
|
||
|
|
from ...tokens import Doc
|
||
|
|
from ...vocab import Vocab
|
||
|
|
from ...matcher import Matcher
|
||
|
|
|
||
|
|
|
||
|
|
def return_false(doc, ent_id, label, start, end):
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def test_matcher_accept():
|
||
|
|
doc = Doc(Vocab(), words=[u'The', u'golf', u'club', u'is', u'broken'])
|
||
|
|
|
||
|
|
golf_pattern = [
|
||
|
|
{ ORTH: "golf"},
|
||
|
|
{ ORTH: "club"}
|
||
|
|
]
|
||
|
|
matcher = Matcher(doc.vocab)
|
||
|
|
|
||
|
|
matcher.add_entity('Sport_Equipment', acceptor=return_false)
|
||
|
|
matcher.add_pattern("Sport_Equipment", golf_pattern)
|
||
|
|
match = matcher(doc)
|
||
|
|
|
||
|
|
assert match == []
|