1  from mlflow.entities import Param
 2  
 3  from tests.helper_functions import random_int, random_str
 4  
 5  
 6  def _check(param, key, value):
 7      assert isinstance(param, Param)
 8      assert param.key == key
 9      assert param.value == value
10  
11  
12  def test_creation_and_hydration():
13      key = random_str(random_int(10, 25))  # random string on size in range [10, 25]
14      value = random_str(random_int(55, 75))  # random string on size in range [55, 75]
15      param = Param(key, value)
16      _check(param, key, value)
17  
18      as_dict = {"key": key, "value": value}
19      assert dict(param) == as_dict
20  
21      proto = param.to_proto()
22      param2 = Param.from_proto(proto)
23      _check(param2, key, value)
24  
25      param3 = Param.from_dictionary(as_dict)
26      _check(param3, key, value)