MLIA-812 Show error message for unsupported functionality0.6.0-rc.2

Change-Id: I68fb8c4e51046e9fc2d91ad8338718ba545209cd

1 files changed, 23 insertions, 1 deletions

diff --git a/tests/test_target_ethos_u_advisor.py b/tests/test_target_ethos_u_advisor.py
index fb68800..11aefc7 100644
--- a/tests/test_target_ethos_u_advisor.py
+++ b/tests/test_target_ethos_u_advisor.py
@@ -1,9 +1,31 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Tests for Ethos-U MLIA module."""
+from pathlib import Path
+
+import pytest
+
+from mlia.core.common import AdviceCategory
+from mlia.core.context import ExecutionContext
+from mlia.target.ethos_u.advisor import configure_and_get_ethosu_advisor
 from mlia.target.ethos_u.advisor import EthosUInferenceAdvisor
 
 
 def test_advisor_metadata() -> None:
     """Test advisor metadata."""
     assert EthosUInferenceAdvisor.name() == "ethos_u_inference_advisor"
+
+
+def test_unsupported_advice_categories(tmp_path: Path, test_tflite_model: Path) -> None:
+    """Test that advisor should throw an exception for unsupported categories."""
+    with pytest.raises(
+        Exception, match="Optimizations are not supported for TensorFlow Lite files."
+    ):
+        ctx = ExecutionContext(
+            output_dir=tmp_path, advice_category={AdviceCategory.OPTIMIZATION}
+        )
+
+        advisor = configure_and_get_ethosu_advisor(
+            ctx, "ethos-u55-256", str(test_tflite_model)
+        )
+        advisor.configure(ctx)