explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-14 11:36:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Allow SPACY_DATA environment variable in website tests

Browse Source
This commit is contained in:
Matthew Honnibal 2015-10-10 17:58:57 +11:00
parent 1cac36bf1c
commit e18fbcb604
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/website/conftest.py
View File

@ -1,11 +1,13 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import pytest import pytest
import os
@pytest.fixture(scope='session') @pytest.fixture(scope='session')
def nlp(): def nlp():
from spacy.en import English from spacy.en import English, LOCAL_DATA_DIR
return English() data_dir = os.environ.get('SPACY_DATA', LOCAL_DATA_DIR)
return English(data_dir=data_dir)
@pytest.fixture() @pytest.fixture()

6
tests/website/test_home.py
View File

@ -1,6 +1,7 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import pytest import pytest
import spacy import spacy
import os
@pytest.fixture() @pytest.fixture()
@ -9,8 +10,9 @@ def token(doc):
def test_load_resources_and_process_text(): def test_load_resources_and_process_text():
from spacy.en import English from spacy.en import English, LOCAL_DATA_DIR
nlp = English() data_dir = os.environ.get('SPACY_DATA', LOCAL_DATA_DIR)
nlp = English(data_dir=data_dir)
doc = nlp('Hello, world. Here are two sentences.') doc = nlp('Hello, world. Here are two sentences.')