aboutsummaryrefslogtreecommitdiff
path: root/tests/test_cli_commands.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_cli_commands.py')
-rw-r--r--tests/test_cli_commands.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/test_cli_commands.py b/tests/test_cli_commands.py
index eaa08e6..fd9e29c 100644
--- a/tests/test_cli_commands.py
+++ b/tests/test_cli_commands.py
@@ -22,19 +22,19 @@ from mlia.devices.ethosu.performance import PerformanceMetrics
from mlia.tools.metadata.common import InstallationManager
-def test_operators_expected_parameters(dummy_context: ExecutionContext) -> None:
+def test_operators_expected_parameters(sample_context: ExecutionContext) -> None:
"""Test operators command wrong parameters."""
with pytest.raises(Exception, match="Model is not provided"):
- operators(dummy_context, "ethos-u55-256")
+ operators(sample_context, "ethos-u55-256")
def test_performance_unknown_target(
- dummy_context: ExecutionContext, test_tflite_model: Path
+ sample_context: ExecutionContext, test_tflite_model: Path
) -> None:
"""Test that command should fail if unknown target passed."""
with pytest.raises(Exception, match="Unable to find target profile unknown"):
performance(
- dummy_context, model=str(test_tflite_model), target_profile="unknown"
+ sample_context, model=str(test_tflite_model), target_profile="unknown"
)
@@ -74,7 +74,7 @@ def test_performance_unknown_target(
],
)
def test_opt_expected_parameters(
- dummy_context: ExecutionContext,
+ sample_context: ExecutionContext,
target_profile: str,
monkeypatch: pytest.MonkeyPatch,
optimization_type: str,
@@ -87,7 +87,7 @@ def test_opt_expected_parameters(
with expected_error:
optimization(
- ctx=dummy_context,
+ ctx=sample_context,
target_profile=target_profile,
model=str(test_keras_model),
optimization_type=optimization_type,
@@ -105,7 +105,7 @@ def test_opt_expected_parameters(
)
def test_opt_valid_optimization_target(
target_profile: str,
- dummy_context: ExecutionContext,
+ sample_context: ExecutionContext,
optimization_type: str,
optimization_target: str,
monkeypatch: pytest.MonkeyPatch,
@@ -115,7 +115,7 @@ def test_opt_valid_optimization_target(
mock_performance_estimation(monkeypatch)
optimization(
- ctx=dummy_context,
+ ctx=sample_context,
target_profile=target_profile,
model=str(test_keras_model),
optimization_type=optimization_type,