wehub-resource-sync
29 lines
1.0 KiB
Python
29 lines
1.0 KiB
Python
from pathlib import Path
|
|
|
|
from clint.config import Config
|
|
from clint.index import SymbolIndex
|
|
from clint.linter import Position, Range, lint_file
|
|
from clint.rules.lazy_module import LazyModule
|
|
|
|
|
|
def test_lazy_module(index: SymbolIndex) -> None:
|
|
# Create a file that looks like mlflow/__init__.py for the rule to apply
|
|
code = """
|
|
from mlflow.utils.lazy_load import LazyLoader
|
|
from typing import TYPE_CHECKING
|
|
|
|
# Bad - LazyLoader module not imported in TYPE_CHECKING block
|
|
anthropic = LazyLoader("mlflow.anthropic", globals(), "mlflow.anthropic")
|
|
|
|
# Good - LazyLoader with corresponding TYPE_CHECKING import
|
|
sklearn = LazyLoader("mlflow.sklearn", globals(), "mlflow.sklearn")
|
|
|
|
if TYPE_CHECKING:
|
|
from mlflow import sklearn # Good - this one is imported
|
|
"""
|
|
config = Config(select={LazyModule.name})
|
|
violations = lint_file(Path("mlflow", "__init__.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, LazyModule) for v in violations)
|
|
assert violations[0].range == Range(Position(5, 12)) # anthropic LazyLoader
|