diff options
| author | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-06-27 09:10:49 +0100 |
|---|---|---|
| committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-06-27 16:18:55 +0100 |
| commit | 0ee201dfa023a9f42e1c8a282e96c231c6769e07 (patch) | |
| tree | abd0048d0f1922470e3fe6befcc32f62d38a52f5 /tests | |
| parent | 7faf2c4763f299ee53b1ed100025ba50021c8313 (diff) | |
| download | mlia-0ee201dfa023a9f42e1c8a282e96c231c6769e07.tar.gz | |
MLIA-522 No 'perf' in mode 'all' for TFLite files
Fix the issue that no performance information is shown for TFLite files
when the mode 'all_tests' is used.
Change-Id: I8b4df4ab84ba9783b582ad449a34bf6177037e14
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/mlia/test_api.py | 2 | ||||
| -rw-r--r-- | tests/mlia/test_core_context.py | 4 | ||||
| -rw-r--r-- | tests/mlia/test_devices_ethosu_advice_generation.py | 5 |
3 files changed, 5 insertions, 6 deletions
diff --git a/tests/mlia/test_api.py b/tests/mlia/test_api.py index 54d4796..09bc509 100644 --- a/tests/mlia/test_api.py +++ b/tests/mlia/test_api.py @@ -55,7 +55,7 @@ def test_get_advice_wrong_category(test_keras_model: Path) -> None: [ "all", ExecutionContext(advice_category=AdviceCategory.PERFORMANCE), - AdviceCategory.PERFORMANCE, + AdviceCategory.ALL, ], [ "all", diff --git a/tests/mlia/test_core_context.py b/tests/mlia/test_core_context.py index 10015aa..44eb976 100644 --- a/tests/mlia/test_core_context.py +++ b/tests/mlia/test_core_context.py @@ -41,7 +41,7 @@ def test_execution_context(tmpdir: str) -> None: ) context_with_default_params = ExecutionContext(working_dir=tmpdir) - assert context_with_default_params.advice_category is None + assert context_with_default_params.advice_category is AdviceCategory.ALL assert context_with_default_params.config_parameters is None assert context_with_default_params.event_handlers is None assert isinstance( @@ -55,7 +55,7 @@ def test_execution_context(tmpdir: str) -> None: expected_str = ( f"ExecutionContext: working_dir={tmpdir}, " - "advice_category=<not set>, " + "advice_category=ALL, " "config_parameters=None, " "verbose=False" ) diff --git a/tests/mlia/test_devices_ethosu_advice_generation.py b/tests/mlia/test_devices_ethosu_advice_generation.py index 98c8a57..5d37376 100644 --- a/tests/mlia/test_devices_ethosu_advice_generation.py +++ b/tests/mlia/test_devices_ethosu_advice_generation.py @@ -2,7 +2,6 @@ # SPDX-License-Identifier: Apache-2.0 """Tests for Ethos-U advice generation.""" from typing import List -from typing import Optional import pytest @@ -387,7 +386,7 @@ def test_ethosu_advice_producer( "advice_category, action_resolver, expected_advice", [ [ - None, + AdviceCategory.ALL, None, [], ], @@ -467,7 +466,7 @@ def test_ethosu_advice_producer( ) def test_ethosu_static_advice_producer( tmpdir: str, - advice_category: Optional[AdviceCategory], + advice_category: AdviceCategory, action_resolver: ActionResolver, expected_advice: List[Advice], ) -> None: |