refactoring: splitting conftest logic into submodules
This commit is contained in:
parent
d39075e85e
commit
babc2a6fc8
@ -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
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user