spaCy/spacy/tests/website/conftest.py

19 lines
390 B
Python
Raw Normal View History

2015-09-28 15:22:13 +03:00
from __future__ import unicode_literals
import pytest
import os
2015-09-28 15:22:13 +03:00
@pytest.fixture(scope='session')
def nlp():
2015-12-07 08:01:28 +03:00
from spacy.en import English
if os.environ.get('SPACY_DATA'):
2016-01-16 14:23:45 +03:00
data_dir = os.environ.get('SPACY_DATA')
else:
2016-01-16 14:23:45 +03:00
data_dir = None
return English(data_dir=data_dir)
2015-09-28 15:22:13 +03:00
@pytest.fixture()
def doc(nlp):
return nlp('Hello, world. Here are two sentences.')