def test_find_triples():
text = "President Obama won the noble prize"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
triples = main.find_triples(tokens)
for triple in triples:
assert (1, 2, 5) == triple
def test_find_triples():
text = "President Obama won the noble prize"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
triples = main.find_triples(tokens)
for triple in triples:
assert (1, 2, 5) == triple
def test_phrase_text_for_head():
text = "A small collection of words"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
assert "words" == main.phrase_text_for_head(tokens, text, 4)
def test_dependents():
text = "I am eating a delicious banana"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
assert [0, 1, 5] == main.dependents(tokens, 2)
assert [3, 4] == main.dependents(tokens, 5)
def test_phrase_text_for_head():
text = "A small collection of words"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
assert "words" == main.phrase_text_for_head(tokens, text, 4)
def test_dependents():
text = "I am eating a delicious banana"
analysis = main.analyze_syntax(text)
tokens = analysis.get('tokens', [])
assert [0, 1, 5] == main.dependents(tokens, 2)
assert [3, 4] == main.dependents(tokens, 5)