def test_max_scope_none(self):
"""Test that if max_scope is not None it will reduce the range
of text which is modified.
"""
doc = self.create_num_target_examples()
assert len(doc.ents) == 3
item = ConTextItem(
"vs", category="UNCERTAIN", rule="BIDIRECTIONAL", max_scope=None
)
tag_object = TagObject(item, 5, 6, doc)
for target in doc.ents:
if tag_object.modifies(target):
tag_object.modify(target)
assert tag_object.num_targets == 3
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")