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()
