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")
