aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-11-24 08:34:38 +0000
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-11-29 14:44:13 +0000
commita34163c9d9a5cc0416bcaea2ebf8383bda9d505c (patch)
tree304c01c607b3a93c250a38df53c417f62196b5fa /tests
parent37959522a805a5e23c930ed79aac84920c3cb208 (diff)
downloadmlia-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.py2
-rw-r--r--tests/test_devices_tosa_data_collection.py2
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(