mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-14 21:57:15 +03:00
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
# coding: utf-8
|
|
|
|
|
|
import re
|
|
from ...pattern.parser import PatternParser
|
|
|
|
|
|
class TestPatternParser:
|
|
def test_empty_query(self):
|
|
assert PatternParser.parse('') is None
|
|
assert PatternParser.parse(' ') is None
|
|
|
|
def test_define_node(self):
|
|
query = "fox [lemma:fox,word:fox]=alias"
|
|
pattern = PatternParser.parse(query)
|
|
|
|
assert pattern is not None
|
|
assert pattern.number_of_nodes() == 1
|
|
assert pattern.number_of_edges() == 0
|
|
|
|
assert 'fox' in pattern.nodes
|
|
|
|
attrs = pattern['fox']
|
|
assert attrs.get('lemma') == 'fox'
|
|
assert attrs.get('word') == 'fox'
|
|
assert attrs.get('_name') == 'fox'
|
|
assert attrs.get('_alias') == 'alias'
|
|
|
|
for adj_list in pattern.adjacency.values():
|
|
assert not adj_list
|
|
|
|
def test_define_node_with_regex(self):
|
|
query = "fox [lemma:/fo.*/]"
|
|
pattern = PatternParser.parse(query)
|
|
|
|
attrs = pattern['fox']
|
|
assert attrs.get('lemma') == re.compile(r'fo.*', re.U)
|
|
|
|
def test_define_edge(self):
|
|
query = "[word:quick] >amod [word:fox]"
|
|
pattern = PatternParser.parse(query)
|
|
|
|
assert pattern is not None
|
|
assert pattern.number_of_nodes() == 2
|
|
assert pattern.number_of_edges() == 1
|
|
|
|
quick_id = [node_id for node_id, node_attr in pattern.nodes.items()
|
|
if node_attr['word'] == 'quick'][0]
|
|
|
|
fox_id = [node_id for node_id, node_attr in pattern.nodes.items()
|
|
if node_attr['word'] == 'fox'][0]
|
|
|
|
quick_map = pattern.adjacency[quick_id]
|
|
fox_map = pattern.adjacency[fox_id]
|
|
|
|
assert len(quick_map) == 0
|
|
assert len(fox_map) == 1
|
|
|
|
dep = fox_map[quick_id]
|
|
|
|
assert dep == 'amod'
|
|
|
|
def test_define_edge_with_regex(self):
|
|
query = "[word:quick] >/amod|nsubj/ [word:fox]"
|
|
pattern = PatternParser.parse(query)
|
|
|
|
quick_id = [node_id for node_id, node_attr in pattern.nodes.items()
|
|
if node_attr['word'] == 'quick'][0]
|
|
|
|
fox_id = [node_id for node_id, node_attr in pattern.nodes.items()
|
|
if node_attr['word'] == 'fox'][0]
|
|
|
|
fox_map = pattern.adjacency[fox_id]
|
|
dep = fox_map[quick_id]
|
|
|
|
assert dep == re.compile(r'amod|nsubj', re.U)
|