spaCy/spacy/tests/website/conftest.py

19 lines
393 B
Python

from __future__ import unicode_literals
import pytest
import os
@pytest.fixture(scope='session')
def nlp():
from spacy.en import English
if os.environ.get('SPACY_DATA'):
data_path = os.environ.get('SPACY_DATA')
else:
data_path = None
return English(data_dir=data_path)
@pytest.fixture()
def doc(nlp):
return nlp('Hello, world. Here are two sentences.')