aboutsummaryrefslogtreecommitdiff
path: root/tests/test_target_cortex_a_operators.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_target_cortex_a_operators.py')
-rw-r--r--tests/test_target_cortex_a_operators.py17
1 files changed, 0 insertions, 17 deletions
diff --git a/tests/test_target_cortex_a_operators.py b/tests/test_target_cortex_a_operators.py
index 8bc48e6..56d6c7b 100644
--- a/tests/test_target_cortex_a_operators.py
+++ b/tests/test_target_cortex_a_operators.py
@@ -2,33 +2,16 @@
# SPDX-License-Identifier: Apache-2.0
"""Tests for Cortex-A operator compatibility."""
from pathlib import Path
-from typing import cast
import pytest
import tensorflow as tf
-from mlia.backend.armnn_tflite_delegate import compat
-from mlia.nn.tensorflow.tflite_graph import TFL_OP
from mlia.nn.tensorflow.utils import convert_to_tflite
from mlia.target.cortex_a.config import CortexAConfiguration
from mlia.target.cortex_a.operators import CortexACompatibilityInfo
from mlia.target.cortex_a.operators import get_cortex_a_compatibility_info
-def test_compat_data() -> None:
- """Make sure all data contains the necessary items."""
- builtin_tfl_ops = {op.name for op in TFL_OP}
- assert "backend" in compat.ARMNN_TFLITE_DELEGATE
- assert "ops" in compat.ARMNN_TFLITE_DELEGATE
-
- ops = cast(dict, compat.ARMNN_TFLITE_DELEGATE["ops"])
- for data in ops.values():
- assert "builtin_ops" in data
- for comp in data["builtin_ops"]:
- assert comp in builtin_tfl_ops
- assert "custom_ops" in data
-
-
def check_get_cortex_a_compatibility_info(
model_path: Path,
expected_success: bool,