diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-11-24 08:34:38 +0000 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-11-29 14:44:13 +0000 |
commit | a34163c9d9a5cc0416bcaea2ebf8383bda9d505c (patch) | |
tree | 304c01c607b3a93c250a38df53c417f62196b5fa /tests | |
parent | 37959522a805a5e23c930ed79aac84920c3cb208 (diff) | |
download | mlia-a34163c9d9a5cc0416bcaea2ebf8383bda9d505c.tar.gz |
Move TOSA checker functions into separate module
- Create module "compat" for tosa_checker backend
- Move TOSA checker functions into new module
- Update tests
Change-Id: Ia07034515fe43b2061b8892535067d21315cc721
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( |