from src.annotation_mapping import map_annotatios def test_map_annotations(): ref_annotations = [(0, 3, "Andrzej"), (7, 11, "psa")] all_annotations = { "A": [(0, 3, "Andrzej"), (7, 11, "psa")], "B": [(0, 3, "AndrzejB"), (7, 11, "psaA")], "C": [(0, 3, "AndrzejC"), (8, 9, "psaC")], } result = map_annotatios(ref_annotations, all_annotations, ["B", "C"]) excepted = { (0, 3, "Andrzej"): {"B": (0, 3, "AndrzejB"), "C": (0, 3, "AndrzejC")}, (7, 11, "psa"): { "B": (7, 11, "psaA"), }, } assert result == excepted