Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

refactoring: splitting conftest logic into submodules

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-14 19:10:15 +02:00
parent d39075e85e
commit babc2a6fc8
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/conftest.py
View File

@ -3,7 +3,6 @@ import logging
import pytest import pytest
from image_prediction.info import Info from image_prediction.info import Info
from image_prediction.pipeline import load_pipeline
from image_prediction.utils import get_logger from image_prediction.utils import get_logger
pytest_plugins = [ pytest_plugins = [
@ -34,7 +33,3 @@ def info_label_map():
return Info return Info
@pytest.fixture
def pipeline():
pipeline = load_pipeline(verbose=False)
return pipeline

6
test/unit_tests/pipeline_test.py
View File

@ -1,3 +1,7 @@
def test_pipeline(pipeline, real_pdf, real_expected_service_response): from image_prediction.pipeline import load_pipeline
def test_pipeline(real_pdf, real_expected_service_response):
pipeline = load_pipeline(verbose=False)
response = list(pipeline(real_pdf)) response = list(pipeline(real_pdf))
assert response == real_expected_service_response assert response == real_expected_service_response