natasha

1

2
import pytest
3

4
from natasha.obj import Name
5

6

7
tests = [
8
    [
9
        'Мустафа Джемилев',
10
        Name('Мустафа', 'Джемилев')
11
    ],
12
    [
13
        'Егор Свиридов',
14
        Name('Егор', 'Свиридов')
15
    ],
16
    [
17
        'Владимир Путин',
18
        Name('Владимир', 'Путин')
19
    ],
20
    [
21
        'Плаксюк Саша',
22
        Name('Саша', 'Плаксюк')
23
    ],
24
    [
25
        'Ищенко Е.П.',
26
        Name('Е', 'Ищенко', 'П')
27
    ],
28
    [
29
        'Фёдор Иванович Шаляпин',
30
        Name('Фёдор', 'Шаляпин', 'Иванович')
31
    ],
32
    [
33
        'Ипполит Матвеевич',
34
        Name('Ипполит', None, 'Матвеевич')
35
    ],
36
    [
37
        'Янукович',
38
        Name(None, 'Янукович')
39
    ],
40
    [
41
        'Авраам',
42
        Name('Авраам')
43
    ],
44
    [
45
        'Гоша Куценко',
46
        Name('Гоша', 'Куценко')
47
    ],
48
    [
49
        'Юрий Георгиевич Куценко',
50
        Name('Юрий', 'Куценко', 'Георгиевич')
51
    ],
52
    [
53
        'Наталья Ищенко',
54
        Name('Наталья', 'Ищенко')
55
    ],
56
    [
57
        'Наталья Сергеевна Ищенко',
58
        Name('Наталья', 'Ищенко', 'Сергеевна')
59
    ],
60
    [
61
        'Монина Нина Гафуровна',
62
        Name('Нина', 'Монина', 'Гафуровна')
63
    ],
64
    [
65
        'В. И. Ленин',
66
        Name('В', 'Ленин', 'И'),
67
    ],
68
]
69

70

71
@pytest.mark.parametrize('test', tests)
72
def test_extractor(names_extractor, test):
73
    text, target = test
74
    pred = names_extractor.find(text).fact
75
    assert pred == target
76