def test_select_covering_also_returns_parent_instances(small_typesystem_xml,
tokens, sentences):
typesystem = load_typesystem(small_typesystem_xml)
SubSentenceType = typesystem.create_type("cassis.SubSentence",
supertypeName="cassis.Sentence")
cas = Cas(typesystem=typesystem)
first_sentence, second_sentence = sentences
annotations = tokens + sentences
subsentence1 = SubSentenceType(begin=first_sentence.begin,
end=first_sentence.end)
subsentence2 = SubSentenceType(begin=second_sentence.begin,
end=second_sentence.end)
annotations.append(subsentence1)
annotations.append(subsentence2)
cas.add_annotations(annotations)
tokens_in_first_sentence = tokens[:6]
tokens_in_second_sentence = tokens[6:]
for token in tokens_in_first_sentence:
result = set(cas.select_covering("cassis.Sentence", token))
assert result == {first_sentence, subsentence1}
for token in tokens_in_second_sentence:
result = set(cas.select_covering("cassis.Sentence", token))
assert result == {second_sentence, subsentence2}
def test_select_covering(small_typesystem_xml, tokens, sentences):
cas = Cas(typesystem=load_typesystem(small_typesystem_xml))
cas.add_annotations(tokens + sentences)
actual_first_sentence, actual_second_sentence = sentences
tokens_in_first_sentence = tokens[:6]
tokens_in_second_sentence = tokens[6:]
for token in tokens_in_first_sentence:
result = list(cas.select_covering("cassis.Sentence", token))
first_sentence = result[0]
assert len(result) == 1
assert actual_first_sentence == first_sentence
for token in tokens_in_second_sentence:
result = list(cas.select_covering("cassis.Sentence", token))
second_sentence = result[0]
assert len(result) == 1
assert actual_second_sentence == second_sentence
