Coverage for mlos_bench/mlos_bench/tests/tunables/tunable_group_subgroup_test.py: 100%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5"""Tests for `TunableGroup.subgroup()` method.""" 

6 

7from mlos_bench.tunables.tunable_groups import TunableGroups 

8 

9 

10def test_tunable_group_subgroup(tunable_groups: TunableGroups) -> None: 

11 """Check that the subgroup() method returns only a selection of tunable 

12 parameters. 

13 """ 

14 tunables = tunable_groups.subgroup(["provision"]) 

15 assert tunables.get_param_values() == {"vmSize": "Standard_B4ms"}