From babc2a6fc8274b77127b2c4ea203d2dbb3bf0d96 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Thu, 14 Apr 2022 19:10:15 +0200 Subject: [PATCH] refactoring: splitting conftest logic into submodules --- test/conftest.py | 5 ----- test/unit_tests/pipeline_test.py | 6 +++++- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/test/conftest.py b/test/conftest.py index c835c54..23add8c 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -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 diff --git a/test/unit_tests/pipeline_test.py b/test/unit_tests/pipeline_test.py index 5431e47..8d2208b 100644 --- a/test/unit_tests/pipeline_test.py +++ b/test/unit_tests/pipeline_test.py @@ -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