test integer keyed params

Author: bpkroth

mlos_bench/mlos_bench/tests/conftest.py

@@ -69,6 +69,17 @@
                 "range": [0, 1000000000]
             }
         }
+    },
+    "integer_keyed": {
+        "cost": 0,
+        "params": {
+            "123": {
+                "description": "A parameter with an integer key",
+                "type": "int",
+                "default": 2,
+                "range": [1, 3]
+            }
+        }
     }
 }
 """

mlos_bench/mlos_bench/tests/tunables/tunable_to_configspace_test.py

@@ -38,12 +38,14 @@ def configuration_space() -> ConfigurationSpace:
         "idle": ["halt", "mwait", "noidle"],
         "kernel_sched_migration_cost_ns": (-1, 500000),
         "kernel_sched_latency_ns": (0, 1000000000),
+        "123": (1, 3),
     })
 
     spaces["vmSize"].default_value = "Standard_B4ms"
     spaces["idle"].default_value = "halt"
     spaces["kernel_sched_migration_cost_ns"].default_value = -1
     spaces["kernel_sched_latency_ns"].default_value = 2000000
+    spaces["123"].default_value = 2
 
     return spaces
 

mlos_bench/mlos_bench/tests/tunables/tunables_str_test.py

@@ -18,6 +18,17 @@ def test_tunable_groups_str(tunable_groups: TunableGroups) -> None:
     """
     # Same as `tunable_groups` (defined in the `conftest.py` file), but in different order:
     tunables_other = TunableGroups({
+        "integer_keyed": {
+            "cost": 0,
+            "params": {
+                "123": {
+                    "description": "A parameter with an integer key",
+                    "type": "int",
+                    "default": 2,
+                    "range": [1, 3]
+                }
+            }
+        },
         "kernel": {
             "cost": 1,
             "params": {