natasha

1

2
import pytest
3

4
from natasha.obj import Date
5

6

7
tests = [
8
    [
9
        '24.01.2017',
10
        Date(2017, 1, 24)
11
    ],
12
    [
13
        '27. 05.99',
14
        Date(1999, 5, 27)
15
    ],
16
    [
17
        '2015 год',
18
        Date(2015)
19
    ],
20
    [
21
        '2014 г',
22
        Date(2014)
23
    ],
24
    [
25
        '1 апреля',
26
        Date(None, 4, 1)
27
    ],
28
    [
29
        'май 2017 г.',
30
        Date(2017, 5)
31
    ],
32
    [
33
        '9 мая 2017 года',
34
        Date(2017, 5, 9)
35
    ],
36
]
37

38

39
@pytest.mark.parametrize('test', tests)
40
def test_extractor(dates_extractor, test):
41
    text, target = test
42
    pred = dates_extractor.find(text).fact
43
    assert pred == target
44