diff options
Diffstat (limited to 'tests/test_nn_tensorflow_config.py')
-rw-r--r-- | tests/test_nn_tensorflow_config.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/test_nn_tensorflow_config.py b/tests/test_nn_tensorflow_config.py index 1a6fbe3..656619d 100644 --- a/tests/test_nn_tensorflow_config.py +++ b/tests/test_nn_tensorflow_config.py @@ -1,4 +1,4 @@ -# 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 config module.""" from contextlib import ExitStack as does_not_raise @@ -45,10 +45,12 @@ def test_convert_tf_to_tflite(tmp_path: Path, test_tf_model: Path) -> None: "test.model", None, pytest.raises( - Exception, - match="The input model format is not supported" - r"\(supported formats: TensorFlow Lite, Keras, " - r"TensorFlow saved model\)!", + ValueError, + match=( + "The input model format is not supported " + r"\(supported formats: TensorFlow Lite, Keras, " + r"TensorFlow saved model\)!" + ), ), ), ], |