import logging

import pytest

from image_prediction.utils import get_logger


pytest_plugins = [
    "test.fixtures.extractor",
    "test.fixtures.image",
    "test.fixtures.image_metadata_pair",
    "test.fixtures.input",
    "test.fixtures.label",
    "test.fixtures.metadata",
    "test.fixtures.model",
    "test.fixtures.model_store",
    "test.fixtures.parameters",
    "test.fixtures.pdf",
    "test.fixtures.target",
]


@pytest.fixture(autouse=True)
def mute_logger():
    logger = get_logger()
    level = logger.level
    logger.setLevel(logging.CRITICAL + 1)
    yield
    logger.setLevel(level)