diff --git a/test/conftest.py b/test/conftest.py index c740a7d..14e408a 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -1,4 +1,5 @@ import json +import logging import os import random import string @@ -36,6 +37,16 @@ from image_prediction.model_loader.loaders.mlflow import MlflowConnector from image_prediction.pipeline import load_pipeline from image_prediction.redai_adapter.mlflow import MlflowModelReader from image_prediction.redai_adapter.model import PredictionModelHandle +from image_prediction.utils import get_logger + + +@pytest.fixture(autouse=True) +def mute_logger(): + logger = get_logger() + level = logger.level + logger.setLevel(logging.CRITICAL + 1) + yield + logger.setLevel(level) @pytest.fixture @@ -431,5 +442,5 @@ def real_expected_service_response(): @pytest.fixture def pipeline(): - pipeline = load_pipeline(verbose=True) + pipeline = load_pipeline(verbose=False) return pipeline diff --git a/test/integration_tests/actual_server_test.py b/test/integration_tests/actual_server_test.py index 31e8431..53ac067 100644 --- a/test/integration_tests/actual_server_test.py +++ b/test/integration_tests/actual_server_test.py @@ -42,7 +42,7 @@ def server(server_type): return make_prediction_server(lambda x: int(x.decode()) // 2) elif server_type == "actual": - return make_prediction_server(lambda x: list(load_pipeline(verbose=True)(x))) + return make_prediction_server(lambda x: list(load_pipeline(verbose=False)(x))) else: raise ValueError(f"Unknown server type {server_type}.") diff --git a/test/unit_tests/mocked_server_test.py b/test/unit_tests/mocked_server_test.py index 4629426..8b8a692 100644 --- a/test/unit_tests/mocked_server_test.py +++ b/test/unit_tests/mocked_server_test.py @@ -1,20 +1,9 @@ import json -import logging import pytest from image_prediction.exceptions import IntentionalTestException from image_prediction.flask import make_prediction_server -from image_prediction.utils import get_logger - - -@pytest.fixture -def mute_logger(): - logger = get_logger() - level = logger.level - logger.setLevel(logging.CRITICAL + 1) - yield - logger.setLevel(level) def predict_fn(x: bytes):