def extract_same_head_lemma(mention_x: MentionDataLight, mention_y: MentionDataLight) -> RelationType: """ Check if input mentions has same head lemma relation Args: mention_x: MentionDataLight mention_y: MentionDataLight Returns: RelationType.SAME_HEAD_LEMMA or RelationType.NO_RELATION_FOUND """ if StringUtils.is_preposition(mention_x.mention_head_lemma) or \ StringUtils.is_preposition(mention_y.mention_head_lemma) or \ StringUtils.is_determiner(mention_x.mention_head_lemma) or \ StringUtils.is_determiner(mention_y.mention_head_lemma): return RelationType.NO_RELATION_FOUND if mention_x.mention_head_lemma == mention_y.mention_head_lemma: return RelationType.SAME_HEAD_LEMMA return RelationType.NO_RELATION_FOUND
def test_is_determiner(): assert StringUtils.is_determiner('the') assert StringUtils.is_determiner('on') is False
def test_is_determiner(): assert StringUtils.is_determiner("the") assert StringUtils.is_determiner("on") is False