Skip to content

Commit dc36303

Browse files
added unit tests for INT parameter type
Signed-off-by: Shashank Mittal <[email protected]>
1 parent b85b4bf commit dc36303

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

test/unit/v1beta1/suggestion/test_hyperopt_service.py

+30
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,36 @@ def test_get_suggestion(self):
213213
parameter_type=api_pb2.DOUBLE,
214214
feasible_space=api_pb2.FeasibleSpace(
215215
max="10", min="5", list=[], step="0.8", distribution=api_pb2.LOG_NORMAL)
216+
),
217+
api_pb2.ParameterSpec(
218+
name="param-11",
219+
parameter_type=api_pb2.INT,
220+
feasible_space=api_pb2.FeasibleSpace(
221+
max="64", min="32", distribution=api_pb2.LOG_NORMAL)
222+
),
223+
api_pb2.ParameterSpec(
224+
name="param-12",
225+
parameter_type=api_pb2.INT,
226+
feasible_space=api_pb2.FeasibleSpace(
227+
max="5", min="1", list=[], distribution=api_pb2.UNIFORM)
228+
),
229+
api_pb2.ParameterSpec(
230+
name="param-13",
231+
parameter_type=api_pb2.INT,
232+
feasible_space=api_pb2.FeasibleSpace(
233+
max="10", min="5", list=[], step="0.8", distribution=api_pb2.UNIFORM)
234+
),
235+
api_pb2.ParameterSpec(
236+
name="param-14",
237+
parameter_type=api_pb2.INT,
238+
feasible_space=api_pb2.FeasibleSpace(
239+
max="10", min="5", list=[], distribution=api_pb2.LOG_UNIFORM)
240+
),
241+
api_pb2.ParameterSpec(
242+
name="param-15",
243+
parameter_type=api_pb2.INT,
244+
feasible_space=api_pb2.FeasibleSpace(
245+
max="100", min="5", list=[], step="0.8", distribution=api_pb2.NORMAL)
216246
)
217247
]
218248
)

0 commit comments

Comments
 (0)