| # SPDX-FileCopyrightText: © 2026 Tenstorrent AI ULC | |
| # SPDX-License-Identifier: Apache-2.0 | |
| """Import-dependency tests for models.common.utility_functions.""" | |
| import ast | |
| from pathlib import Path | |
| def test_pytest_is_not_imported_at_module_scope(): | |
| utility_functions_path = Path(__file__).parents[2] / "utility_functions.py" | |
| syntax_tree = ast.parse(utility_functions_path.read_text()) | |
| top_level_pytest_imports = [ | |
| node | |
| for node in syntax_tree.body | |
| if (isinstance(node, ast.Import) and any(alias.name == "pytest" for alias in node.names)) | |
| or (isinstance(node, ast.ImportFrom) and node.module == "pytest") | |
| ] | |
| assert not top_level_pytest_imports, "pytest must remain an optional test-only dependency" | |
| def test_ti_skip_imports_pytest_lazily(): | |
| utility_functions_path = Path(__file__).parents[2] / "utility_functions.py" | |
| syntax_tree = ast.parse(utility_functions_path.read_text()) | |
| ti_skip = next(node for node in syntax_tree.body if isinstance(node, ast.FunctionDef) and node.name == "ti_skip") | |
| assert any( | |
| isinstance(node, ast.Import) and any(alias.name == "pytest" for alias in node.names) for node in ti_skip.body | |
| ), "ti_skip must import pytest when the test helper is used" | |