def test_handle_resolution_overlap_with_overlap():
store = ModelStore()
resp = store.handle_resolution_overlap([{
'start': 0,
'end': 2,
'entity': 'karan singla'
}, {
'start': 1,
'end': 3,
'entity': 'singla is'
}, {
'start': 2,
'end': 6,
'entity': 'is famous in Delhi'
}])
expected = [{
'start': 0,
'end': 2,
'entity': 'karan singla'
}, {
'start': 2,
'end': 6,
'entity': 'is famous in Delhi'
}]
assert resp == expected
def test_handle_resolution_overlap_without_overlap():
store = ModelStore()
resp = store.handle_resolution_overlap([{
'start': 4,
'end': 6
}, {
'start': 6,
'end': 7
}, {
'start': 8,
'end': 10
}])
expected = [{
'start': 4,
'end': 6
}, {
'start': 6,
'end': 7
}, {
'start': 8,
'end': 10
}]
assert resp == expected
def test_handle_resolution_overlap_with_empty_list():
store = ModelStore()
resp = store.handle_resolution_overlap([])
expected = []
assert resp == expected