added unit tests for INT parameter type

Signed-off-by: Shashank Mittal <[email protected]>
kubeflow · Sep 22, 2024 · dc36303 · dc36303
1 parent b85b4bf
commit dc36303
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/test/unit/v1beta1/suggestion/test_hyperopt_service.py b/test/unit/v1beta1/suggestion/test_hyperopt_service.py
@@ -213,6 +213,36 @@ def test_get_suggestion(self):
  parameter_type=api_pb2.DOUBLE,
  feasible_space=api_pb2.FeasibleSpace(
  max="10", min="5", list=[], step="0.8", distribution=api_pb2.LOG_NORMAL)
+ ),
+ api_pb2.ParameterSpec(
+ name="param-11",
+ parameter_type=api_pb2.INT,
+ feasible_space=api_pb2.FeasibleSpace(
+ max="64", min="32", distribution=api_pb2.LOG_NORMAL)
+ ),
+ api_pb2.ParameterSpec(
+ name="param-12",
+ parameter_type=api_pb2.INT,
+ feasible_space=api_pb2.FeasibleSpace(
+ max="5", min="1", list=[], distribution=api_pb2.UNIFORM)
+ ),
+ api_pb2.ParameterSpec(
+ name="param-13",
+ parameter_type=api_pb2.INT,
+ feasible_space=api_pb2.FeasibleSpace(
+ max="10", min="5", list=[], step="0.8", distribution=api_pb2.UNIFORM)
+ ),
+ api_pb2.ParameterSpec(
+ name="param-14",
+ parameter_type=api_pb2.INT,
+ feasible_space=api_pb2.FeasibleSpace(
+ max="10", min="5", list=[], distribution=api_pb2.LOG_UNIFORM)
+ ),
+ api_pb2.ParameterSpec(
+ name="param-15",
+ parameter_type=api_pb2.INT,
+ feasible_space=api_pb2.FeasibleSpace(
+ max="100", min="5", list=[], step="0.8", distribution=api_pb2.NORMAL)
  )
  ]
  )