1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Tests for the module mixins."""
import pytest
from mlia.core.common import AdviceCategory
from mlia.core.context import Context
from mlia.core.context import ExecutionContext
from mlia.core.mixins import ContextMixin
from mlia.core.mixins import ParameterResolverMixin
def test_context_mixin(dummy_context: Context) -> None:
"""Test ContextMixin."""
class SampleClass(ContextMixin):
"""Sample class."""
sample_object = SampleClass()
sample_object.set_context(dummy_context)
assert sample_object.context == dummy_context
class TestParameterResolverMixin:
"""Tests for parameter resolver mixin."""
@staticmethod
def test_parameter_resolver_mixin(dummy_context: ExecutionContext) -> None:
"""Test ParameterResolverMixin."""
class SampleClass(ParameterResolverMixin):
"""Sample class."""
def __init__(self) -> None:
"""Init sample object."""
self.context = dummy_context
self.context.update(
advice_category=AdviceCategory.OPERATORS,
event_handlers=[],
config_parameters={"section": {"param": 123}},
)
sample_object = SampleClass()
value = sample_object.get_parameter("section", "param")
assert value == 123
with pytest.raises(
Exception, match="Parameter param expected to have type <class 'str'>"
):
value = sample_object.get_parameter("section", "param", expected_type=str)
with pytest.raises(Exception, match="Parameter no_param is not set"):
value = sample_object.get_parameter("section", "no_param")
@staticmethod
def test_parameter_resolver_mixin_no_config(
dummy_context: ExecutionContext,
) -> None:
"""Test ParameterResolverMixin without config params."""
class SampleClassNoConfig(ParameterResolverMixin):
"""Sample context without config params."""
def __init__(self) -> None:
"""Init sample object."""
self.context = dummy_context
with pytest.raises(Exception, match="Configuration parameters are not set"):
sample_object_no_config = SampleClassNoConfig()
sample_object_no_config.get_parameter("section", "param", expected_type=str)
@staticmethod
def test_parameter_resolver_mixin_bad_section(
dummy_context: ExecutionContext,
) -> None:
"""Test ParameterResolverMixin without config params."""
class SampleClassBadSection(ParameterResolverMixin):
"""Sample context with bad section in config."""
def __init__(self) -> None:
"""Init sample object."""
self.context = dummy_context
self.context.update(
advice_category=AdviceCategory.OPERATORS,
event_handlers=[],
config_parameters={"section": ["param"]},
)
with pytest.raises(
Exception,
match="Parameter section section has wrong format, "
"expected to be a dictionary",
):
sample_object_bad_section = SampleClassBadSection()
sample_object_bad_section.get_parameter(
"section", "param", expected_type=str
)
|
