diff options
Diffstat (limited to 'tests/test_target_ethos_u_performance.py')
-rw-r--r-- | tests/test_target_ethos_u_performance.py | 16 |
1 files changed, 1 insertions, 15 deletions
diff --git a/tests/test_target_ethos_u_performance.py b/tests/test_target_ethos_u_performance.py index 76860b5..3042265 100644 --- a/tests/test_target_ethos_u_performance.py +++ b/tests/test_target_ethos_u_performance.py @@ -1,24 +1,10 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Performance estimation tests.""" from unittest.mock import MagicMock import pytest -from mlia.target.ethos_u.performance import MemorySizeType -from mlia.target.ethos_u.performance import MemoryUsage - - -def test_memory_usage_conversion() -> None: - """Test MemoryUsage objects conversion.""" - memory_usage_in_kb = MemoryUsage(1, 2, 3, 4, 5, MemorySizeType.KILOBYTES) - assert memory_usage_in_kb.in_kilobytes() == memory_usage_in_kb - - memory_usage_in_bytes = MemoryUsage( - 1 * 1024, 2 * 1024, 3 * 1024, 4 * 1024, 5 * 1024 - ) - assert memory_usage_in_bytes.in_kilobytes() == memory_usage_in_kb - def mock_performance_estimation(monkeypatch: pytest.MonkeyPatch) -> None: """Mock performance estimation.""" |