natasha

1

2
import pytest
3

4
from natasha.obj import (
5
    AddrPart as Part,
6
    Addr
7
)
8

9

10
tests = [
11
    [
12
        'Россия, Вологодская обл. г. Череповец, пр.Победы 93 б',
13
        Addr([
14
            Part('Россия', 'страна'),
15
            Part('Вологодская', 'область'),
16
            Part('Череповец', 'город'),
17
            Part('Победы', 'проезд'),
18
        ])
19
    ],
20
    [
21
        '692909, РФ, Приморский край, г. Находка, ул. Добролюбова, 18',
22
        Addr([
23
            Part('692909', 'индекс'),
24
            Part('РФ', 'страна'),
25
            Part('Приморский', 'край'),
26
            Part('Находка', 'город'),
27
            Part('Добролюбова', 'улица'),
28
        ])
29
    ],
30
    [
31
        'д. Федоровка, ул. Дружбы, 13',
32
        Addr([
33
            Part('Федоровка', 'деревня'),
34
            Part('Дружбы', 'улица'),
35
        ])
36
    ],
37
    [
38
        'Россия, 129110, г.Москва, Олимпийский проспект, 22',
39
        Addr([
40
            Part('Россия', 'страна'),
41
            Part('129110', 'индекс'),
42
            Part('Москва', 'город'),
43
            Part('Олимпийский', 'проспект'),
44
        ])
45
    ],
46
    [
47
        'г. Санкт-Петербург, Красногвардейский пер., д. 15',
48
        Addr([
49
            Part('Санкт-Петербург', 'город'),
50
            Part('Красногвардейский', 'переулок'),
51
            Part('15', 'дом')
52
        ])
53
    ],
54
    [
55
        'Республика Карелия,г.Петрозаводск,ул.Маршала Мерецкова, д.8 Б,офис 4',
56
        Addr([
57
            Part('Карелия', 'республика'),
58
            Part('Петрозаводск', 'город'),
59
            Part('Маршала Мерецкова', 'улица'),
60
            Part('8 Б', 'дом'),
61
            Part('4', 'офис')
62
        ])
63
    ],
64
    [
65
        '628000, ХМАО-Югра, г.Ханты-Мансийск, ул. Ледовая , д.19',
66
        Addr([
67
            Part('628000', 'индекс'),
68
            Part('ХМАО-Югра'),
69
            Part('Ханты-Мансийск', 'город'),
70
            Part('Ледовая', 'улица'),
71
            Part('19', 'дом')
72
        ])
73
    ],
74
    [
75
        'ХМАО г.Нижневартовск пер.Ягельный 17',
76
        Addr([
77
            Part('ХМАО'),
78
            Part('Нижневартовск', 'город'),
79
            Part('Ягельный', 'переулок'),
80
        ])
81
    ],
82
    [
83
        'Белгородская обл, пгт Борисовка,ул. Рудого д.160',
84
        Addr([
85
            Part('Белгородская', 'область'),
86
            Part('Борисовка', 'посёлок'),
87
            Part('Рудого', 'улица'),
88
            Part('160', 'дом')
89
        ])
90
    ],
91
    [
92
        'Самарская область, п.г.т. Алексеевка, ул. Ульяновская д. 21',
93
        Addr([
94
            Part('Самарская', 'область'),
95
            Part('Алексеевка', 'посёлок'),
96
            Part('Ульяновская', 'улица'),
97
            Part('21', 'дом')
98
        ])
99

100
    ],
101
    [
102
        'Мурманская обл поселок городского типа Молочный, ул.Гальченко д.11',
103
        Addr([
104
            Part('Мурманская', 'область'),
105
            Part('Молочный', 'посёлок'),
106
            Part('Гальченко', 'улица'),
107
            Part('11', 'дом')
108
        ])
109
    ],
110
    [
111
        'ул. Народного Ополчения д. 9к.3',
112
        Addr([
113
            Part('Народного Ополчения', 'улица'),
114
            Part('9к', 'дом'),
115
        ])
116
    ],
117
    [
118
        'ул. Б. Пироговская, д.37/430',
119
        Addr([
120
            Part('Б. Пироговская', 'улица'),
121
            Part('37/430', 'дом')
122
        ])
123
    ],
124
    [
125
        'Ставропольский край, Изобильненский район, город Изобильный, улица Чапаева, дом 68',
126
        Addr([
127
            Part('Ставропольский', 'край'),
128
            Part('Изобильненский', 'район'),
129
            Part('Изобильный', 'город'),
130
            Part('Чапаева', 'улица'),
131
            Part('68', 'дом')
132
        ])
133
    ],
134
]
135

136

137
@pytest.mark.parametrize('test', tests)
138
def test_extractor(addr_extractor, test):
139
    text, target = test
140
    pred = addr_extractor.find(text).fact
141
    assert pred == target
142