From 0ee201dfa023a9f42e1c8a282e96c231c6769e07 Mon Sep 17 00:00:00 2001
From: Benjamin Klimczak <benjamin.klimczak@arm.com>
Date: Mon, 27 Jun 2022 09:10:49 +0100
Subject: 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
---
 tests/mlia/test_api.py                              | 2 +-
 tests/mlia/test_core_context.py                     | 4 ++--
 tests/mlia/test_devices_ethosu_advice_generation.py | 5 ++---
 3 files changed, 5 insertions(+), 6 deletions(-)

(limited to 'tests/mlia')

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:
-- 
cgit v1.2.1