def test_get_hyperparameters_flat(self): block_1 = Mock() block_1.get_hyperparameters.return_value = { 'a': 'a' } block_2 = Mock() block_2.get_hyperparameters.return_value = { 'b': 'b', 'c': 'c', } blocks = OrderedDict(( ('a.primitive.Name#1', block_1), ('a.primitive.Name#2', block_2), )) mlpipeline = MLPipeline(['a_primitive']) mlpipeline.blocks = blocks hyperparameters = mlpipeline.get_hyperparameters(flat=True) assert hyperparameters == { ('a.primitive.Name#1', 'a'): 'a', ('a.primitive.Name#2', 'b'): 'b', ('a.primitive.Name#2', 'c'): 'c', } block_1.get_hyperparameters.assert_called_once_with() block_2.get_hyperparameters.assert_called_once_with()
def test_get_hyperparameters(self): block_1 = Mock() block_1.get_hyperparameters.return_value = {'a': 'a'} block_2 = Mock() block_2.get_hyperparameters.return_value = { 'b': 'b', 'c': 'c', } blocks = OrderedDict(( ('a.primitive.Name#1', block_1), ('a.primitive.Name#2', block_2), )) mlpipeline = MLPipeline(list()) mlpipeline.blocks = blocks hyperparameters = mlpipeline.get_hyperparameters() assert hyperparameters == { 'a.primitive.Name#1': { 'a': 'a', }, 'a.primitive.Name#2': { 'b': 'b', 'c': 'c', }, } block_1.get_hyperparameters.assert_called_once_with() block_2.get_hyperparameters.assert_called_once_with()