def test_get_head(): token1 = Token('I', 0) token2 = Token('love', 1, 0) token3 = Token('Berlin', 2, 1) sentence = Sentence() sentence.add_token(token1) sentence.add_token(token2) sentence.add_token(token3) assert (token2 == token3.get_head()) assert (token1 == token2.get_head()) assert (None == token1.get_head())
def test_get_head(): token1 = Token("I", 0) token2 = Token("love", 1, 0) token3 = Token("Berlin", 2, 1) sentence: Sentence = Sentence() sentence.add_token(token1) sentence.add_token(token2) sentence.add_token(token3) assert token2 == token3.get_head() assert token1 == token2.get_head() assert None == token1.get_head()