muting logger in tests

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

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}.")

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):