def test_max_targets_none(self): """Check that if max_targets is None it will not reduce the targets to the two closest ents. """ doc = self.create_num_target_examples() assert len(doc.ents) == 3 item = ConTextItem( "vs", category="UNCERTAIN", rule="BIDIRECTIONAL", max_targets=None ) # Set "vs" to be the modifier tag_object = TagObject(item, 5, 6, doc) for target in doc.ents: tag_object.modify(target) assert tag_object.num_targets == 3 tag_object.reduce_targets() assert tag_object.num_targets == 3
def test_max_targets_less_than_targets(self): """Check that if max_targets is not None it will reduce the targets to the two closest ents. """ doc = self.create_num_target_examples() assert len(doc.ents) == 3 item = ConTextItem( "vs", category="UNCERTAIN", rule="BIDIRECTIONAL", max_targets=2 ) # Set "vs" to be the modifier tag_object = TagObject(item, 5, 6, doc) for target in doc.ents: tag_object.modify(target) assert tag_object.num_targets == 3 tag_object.reduce_targets() assert tag_object.num_targets == 2 for target in tag_object._targets: assert target.lower_ in ("pneumonia", "copd")