Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • bump-prettytable
  • bump-edlib
  • bump-editdistance
  • ruff-a
  • 0.3.3
  • 0.3.3rc3
  • 0.3.3rc2
  • 0.3.3-rc1
  • 0.3.2
  • 0.3.1
  • 0.3.1-rc1
12 results
Blame Permalink
test_get_labels_aligned.py 2.67 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119 






import pytest

from nerval import evaluate

fake_annot_original = "Gérard de Nerval was born in Paris in 1808 ."
fake_predict_original = "G*rard de *N*erval bo*rn in Paris in 1833 *."

fake_annot_aligned = "Gérard de -N-erval was bo-rn in Paris in 1808 -."
fake_predict_aligned = "G*rard de *N*erval ----bo*rn in Paris in 1833 *."

# fmt: off
fake_annot_tags_original = [
    "B-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "O",
    "O", "O", "O",
    "O",
    "O", "O", "O", "O",
    "O",
    "O", "O",
    "O",
    "B-LOC", "I-LOC", "I-LOC", "I-LOC", "I-LOC",
    "O",
    "O", "O",
    "O",
    "B-DAT", "I-DAT", "I-DAT", "I-DAT",
    "O",
    "O",
]

fake_predict_tags_original = [
    "B-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "O",
    "O", "O", "O", "O", "O",
    "O",
    "O", "O",
    "O",
    "***", "***", "***", "***", "***",
    "O",
    "O", "O",
    "O",
    "B-DAT", "I-DAT", "I-DAT", "I-DAT",
    "O",
    "O", "O",
]

expected_annot_tags_aligned = [
    "B-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "O",
    "O", "O", "O",
    "O",
    "O", "O", "O", "O", "O",
    "O",
    "O", "O",
    "O",
    "B-LOC", "I-LOC", "I-LOC", "I-LOC", "I-LOC",
    "O",
    "O", "O",
    "O",
    "B-DAT", "I-DAT", "I-DAT", "I-DAT",
    "O",
    "O", "O",
]

expected_predict_tags_aligned = [
    "B-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER",
    "I-PER",
    "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER", "I-PER",
    "O",
    "O", "O", "O", "O",
    "O", "O", "O", "O", "O",
    "O",
    "O", "O",
    "O",
    "***", "***", "***", "***", "***",
    "O",
    "O", "O",
    "O",
    "B-DAT", "I-DAT", "I-DAT", "I-DAT",
    "O",
    "O", "O",
]
# fmt: on


@pytest.mark.parametrize(
    ("test_input", "expected"),
    [
        (
            (fake_annot_original, fake_annot_aligned, fake_annot_tags_original),
            expected_annot_tags_aligned,
        ),
        (
            (fake_predict_original, fake_predict_aligned, fake_predict_tags_original),
            expected_predict_tags_aligned,
        ),
    ],
)
def test_get_labels_aligned(test_input, expected):
    assert evaluate.get_labels_aligned(*test_input) == expected


def test_get_labels_aligned_empty_entry():
    with pytest.raises(AssertionError):
        evaluate.get_labels_aligned(None, None, None)