test_skinny_client_omits_data_science_libs.py

import os

import pytest


@pytest.fixture(autouse=True)
def is_skinny():
    if "MLFLOW_SKINNY" not in os.environ:
        pytest.skip("This test is only valid for the skinny client")


def test_fails_import_flask():
    import mlflow  # noqa: F401

    with pytest.raises(ImportError, match="flask"):
        import flask  # noqa: F401


def test_fails_import_pandas():
    import mlflow  # noqa: F401

    with pytest.raises(ImportError, match="pandas"):
        import pandas  # noqa: F401


def test_fails_import_numpy():
    import mlflow  # noqa: F401

    with pytest.raises(ImportError, match="numpy"):
        import numpy  # noqa: F401