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""" 

6Tests for `TunableGroup.subgroup()` method. 

7""" 

8 

9from mlos_bench.tunables.tunable_groups import TunableGroups 

10 

11 

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

13 """ 

14 Check that the subgroup() method returns only a selection of tunable parameters. 

15 """ 

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

17 assert tunables.get_param_values() == {'vmSize': 'Standard_B4ms'}