refactoring: splitting conftest logic into submodules

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

View File

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

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))
assert response == real_expected_service_response