aboutsummaryrefslogtreecommitdiff
path: root/tests/test_target_ethos_u_performance.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_target_ethos_u_performance.py')
-rw-r--r--tests/test_target_ethos_u_performance.py16
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."""