explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-03-26 12:54:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix fuzzy unit tests

Browse Source
This commit is contained in:
Raphaël Bournhonesque 2017-07-01 15:03:32 +02:00
parent f4748834d9
commit 8592f3de47
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
spacy/tests/pattern/parser.py
View File

@ -44,25 +44,33 @@ class TestPatternParser:
assert pattern.number_of_nodes() == 2 assert pattern.number_of_nodes() == 2
assert pattern.number_of_edges() == 1 assert pattern.number_of_edges() == 1
base_node_id = list(pattern.adjacency.keys())[0] quick_id = [node_id for node_id, node_attr in pattern.nodes.items()
adj_map = pattern.adjacency[base_node_id] if node_attr['word'] == 'quick'][0]
assert len(adj_map) == 1 fox_id = [node_id for node_id, node_attr in pattern.nodes.items()
head_node_id = list(adj_map.keys())[0] if node_attr['word'] == 'fox'][0]
dep = adj_map[head_node_id]
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' assert dep == 'amod'
assert pattern[base_node_id]['word'] == 'fox'
assert pattern[head_node_id]['word'] == 'quick'
def test_define_edge_with_regex(self): def test_define_edge_with_regex(self):
query = "[word:quick] >/amod|nsubj/ [word:fox]" query = "[word:quick] >/amod|nsubj/ [word:fox]"
pattern = PatternParser.parse(query) pattern = PatternParser.parse(query)
base_node_id = list(pattern.adjacency.keys())[0] quick_id = [node_id for node_id, node_attr in pattern.nodes.items()
adj_map = pattern.adjacency[base_node_id] 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 len(adj_map) == 1
head_node_id = list(adj_map.keys())[0]
dep = adj_map[head_node_id]
assert dep == re.compile(r'amod|nsubj', re.U) assert dep == re.compile(r'amod|nsubj', re.U)