blob: 41e977f6b3be7aefbd929f823e6e87706a6a953a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Tests for TOSA data analysis module."""
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.target.tosa.data_analysis import ModelIsNotTOSACompatible
from mlia.target.tosa.data_analysis import ModelIsTOSACompatible
from mlia.target.tosa.data_analysis import TOSADataAnalyzer
@pytest.mark.parametrize(
"input_data, expected_facts",
[
[
TOSACompatibilityInfo(True, []),
[ModelIsTOSACompatible()],
],
[
TOSACompatibilityInfo(False, []),
[ModelIsNotTOSACompatible()],
],
],
)
def test_tosa_data_analyzer(input_data: DataItem, expected_facts: list[Fact]) -> None:
"""Test TOSA data analyzer."""
analyzer = TOSADataAnalyzer()
analyzer.analyze_data(input_data)
assert analyzer.get_analyzed_data() == expected_facts
|