From 70860214b384c663d1712926be6e168f51892c6e Mon Sep 17 00:00:00 2001 From: Brian Kroth Date: Fri, 4 Oct 2024 21:34:07 +0000 Subject: [PATCH] a little more tests --- .../tests/tunables/tunables_assign_test.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/mlos_bench/mlos_bench/tests/tunables/tunables_assign_test.py b/mlos_bench/mlos_bench/tests/tunables/tunables_assign_test.py index c148101c90..e4ddbed28b 100644 --- a/mlos_bench/mlos_bench/tests/tunables/tunables_assign_test.py +++ b/mlos_bench/mlos_bench/tests/tunables/tunables_assign_test.py @@ -37,11 +37,19 @@ def test_tunables_assign_defaults(tunable_groups: TunableGroups) -> None: tunable_groups.assign({}).reset() assert tunable_groups_defaults == tunable_groups assert tunable_groups.is_defaults() - tunable_groups.assign({"vmSize": "Standard_B2s"}).reset() + new_vm_size = "Standard_B2s" + assert tunable_groups["vmSize"] != new_vm_size + # Change one value. + tunable_groups.assign({"vmSize": new_vm_size}).reset() + assert tunable_groups["vmSize"] == new_vm_size + # Check that the other values are still defaults. + idle_tunable, _ = tunable_groups.get_tunable("idle") + assert idle_tunable.is_default() assert tunable_groups_defaults != tunable_groups assert not tunable_groups.is_defaults() + # Reassign defaults. tunable_groups.assign({}).reset() - assert tunable_groups["vmSize"] != "Standard_B2s" + assert tunable_groups["vmSize"] != new_vm_size assert tunable_groups.is_defaults() assert tunable_groups_defaults == tunable_groups