def test_should_deduplicate_overlapping_slots(self):
# Given
language = LANGUAGE_EN
slots = [
unresolved_slot([0, 3], "kid", "e", "s1"),
unresolved_slot([4, 8], "loco", "e1", "s2"),
unresolved_slot([0, 8], "kid loco", "e1", "s3"),
unresolved_slot([9, 13], "song", "e2", "s4"),
]
# When
deduplicated_slots = _deduplicate_overlapping_slots(slots, language)
# Then
expected_slots = [
unresolved_slot([0, 8], "kid loco", "e1", "s3"),
unresolved_slot([9, 13], "song", "e2", "s4"),
]
self.assertSequenceEqual(deduplicated_slots, expected_slots)
def test_should_deduplicate_overlapping_slots(self):
# Given
language = LANGUAGE_EN
slots = [
unresolved_slot([3, 7], "non_overlapping1", "e", "s1"),
unresolved_slot([9, 16], "aaaaaaa", "e1", "s2"),
unresolved_slot([10, 18], "bbbbbbbb", "e1", "s3"),
unresolved_slot([17, 23], "b cccc", "e1", "s4"),
unresolved_slot([50, 60], "non_overlapping2", "e", "s5"),
]
# When
deduplicated_slots = _deduplicate_overlapping_slots(slots, language)
# Then
expected_slots = [
unresolved_slot([3, 7], "non_overlapping1", "e", "s1"),
unresolved_slot([17, 23], "b cccc", "e1", "s4"),
unresolved_slot([50, 60], "non_overlapping2", "e", "s5"),
]
self.assertSequenceEqual(deduplicated_slots, expected_slots)