diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-12-13 09:39:21 +0000 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-12-16 09:25:21 +0000 |
commit | 8da6441c87532d20d3c8dad538e46dc99d073927 (patch) | |
tree | 6afb7f0c82c5c357f7497413f0e18473e84d472c /tests/test_backend_tosa_compat.py | |
parent | ba9aeace4c3d3f4f31830d68b5cc7dd3f4bf1dde (diff) | |
download | mlia-8da6441c87532d20d3c8dad538e46dc99d073927.tar.gz |
MLIA-460 Revisit backend management
- Provide command for backend installation in case
if backend is not available
- Fix issue with connection timeout during downloading
- Show installation tools output only in verbose mode
Change-Id: Ic0e495ba19879cc2cda4fd0bce20b57ba896cfeb
Diffstat (limited to 'tests/test_backend_tosa_compat.py')
-rw-r--r-- | tests/test_backend_tosa_compat.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/test_backend_tosa_compat.py b/tests/test_backend_tosa_compat.py index 4c4dc5a..52aaa44 100644 --- a/tests/test_backend_tosa_compat.py +++ b/tests/test_backend_tosa_compat.py @@ -10,6 +10,7 @@ from unittest.mock import MagicMock import pytest +from mlia.backend.errors import BackendUnavailableError from mlia.backend.tosa_checker.compat import get_tosa_compatibility_info from mlia.backend.tosa_checker.compat import Operator from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo @@ -31,7 +32,9 @@ def test_compatibility_check_should_fail_if_checker_not_available( """Test that compatibility check should fail if TOSA checker is not available.""" replace_get_tosa_checker_with_mock(monkeypatch, None) - with pytest.raises(Exception, match="TOSA checker is not available"): + with pytest.raises( + BackendUnavailableError, match="Backend tosa-checker is not available" + ): get_tosa_compatibility_info(test_tflite_model) |