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)