mlflow--mlflow
wehub-resource-sync
31 行
647 B
Python
31 行
647 B
Python
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
|