natasha

Форк
0
/
conftest.py 
66 строк · 1.2 Кб
1

2
import pytest
3

4
from natasha import (
5
    Segmenter,
6
    MorphVocab,
7

8
    NewsEmbedding,
9
    NewsMorphTagger,
10
    NewsSyntaxParser,
11
    NewsNERTagger,
12

13
    NamesExtractor,
14
    DatesExtractor,
15
    MoneyExtractor,
16
    AddrExtractor,
17
)
18

19

20
@pytest.fixture(scope='session')
21
def segmenter():
22
    return Segmenter()
23

24

25
@pytest.fixture(scope='session')
26
def morph_vocab():
27
    return MorphVocab()
28

29

30
@pytest.fixture(scope='session')
31
def embedding():
32
    return NewsEmbedding()
33

34

35
@pytest.fixture(scope='session')
36
def morph_tagger(embedding):
37
    return NewsMorphTagger(embedding)
38

39

40
@pytest.fixture(scope='session')
41
def syntax_parser(embedding):
42
    return NewsSyntaxParser(embedding)
43

44

45
@pytest.fixture(scope='session')
46
def ner_tagger(embedding):
47
    return NewsNERTagger(embedding)
48

49

50
@pytest.fixture(scope='session')
51
def names_extractor(morph_vocab):
52
    return NamesExtractor(morph_vocab)
53

54

55
@pytest.fixture(scope='session')
56
def dates_extractor(morph_vocab):
57
    return DatesExtractor(morph_vocab)
58

59

60
@pytest.fixture(scope='session')
61
def money_extractor(morph_vocab):
62
    return MoneyExtractor(morph_vocab)
63

64

65
@pytest.fixture(scope='session')
66
def addr_extractor(morph_vocab):
67
    return AddrExtractor(morph_vocab)
68

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.