diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_backend_tosa_compat.py (renamed from tests/test_devices_tosa_operators.py) | 9 | ||||
-rw-r--r-- | tests/test_devices_tosa_data_analysis.py | 2 | ||||
-rw-r--r-- | tests/test_devices_tosa_data_collection.py | 2 |
3 files changed, 7 insertions, 6 deletions
diff --git a/tests/test_devices_tosa_operators.py b/tests/test_backend_tosa_compat.py index d4372aa..4c4dc5a 100644 --- a/tests/test_devices_tosa_operators.py +++ b/tests/test_backend_tosa_compat.py @@ -10,9 +10,9 @@ from unittest.mock import MagicMock import pytest -from mlia.devices.tosa.operators import get_tosa_compatibility_info -from mlia.devices.tosa.operators import Operator -from mlia.devices.tosa.operators import TOSACompatibilityInfo +from mlia.backend.tosa_checker.compat import get_tosa_compatibility_info +from mlia.backend.tosa_checker.compat import Operator +from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo def replace_get_tosa_checker_with_mock( @@ -20,7 +20,8 @@ def replace_get_tosa_checker_with_mock( ) -> None: """Replace TOSA checker with mock.""" monkeypatch.setattr( - "mlia.devices.tosa.operators.get_tosa_checker", MagicMock(return_value=mock) + "mlia.backend.tosa_checker.compat.get_tosa_checker", + MagicMock(return_value=mock), ) diff --git a/tests/test_devices_tosa_data_analysis.py b/tests/test_devices_tosa_data_analysis.py index ff95978..f2da691 100644 --- a/tests/test_devices_tosa_data_analysis.py +++ b/tests/test_devices_tosa_data_analysis.py @@ -5,12 +5,12 @@ from __future__ import annotations import pytest +from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo from mlia.core.common import DataItem from mlia.core.data_analysis import Fact from mlia.devices.tosa.data_analysis import ModelIsNotTOSACompatible from mlia.devices.tosa.data_analysis import ModelIsTOSACompatible from mlia.devices.tosa.data_analysis import TOSADataAnalyzer -from mlia.devices.tosa.operators import TOSACompatibilityInfo @pytest.mark.parametrize( diff --git a/tests/test_devices_tosa_data_collection.py b/tests/test_devices_tosa_data_collection.py index b9c0b4c..0c1eda1 100644 --- a/tests/test_devices_tosa_data_collection.py +++ b/tests/test_devices_tosa_data_collection.py @@ -6,9 +6,9 @@ from unittest.mock import MagicMock import pytest +from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo from mlia.core.context import ExecutionContext from mlia.devices.tosa.data_collection import TOSAOperatorCompatibility -from mlia.devices.tosa.operators import TOSACompatibilityInfo def test_tosa_data_collection( |