def test_set_hyperparameters_flat(self): block_1 = Mock() block_2 = Mock() blocks = OrderedDict(( ('a.primitive.Name#1', block_1), ('a.primitive.Name#2', block_2), )) mlpipeline = MLPipeline(list()) mlpipeline.blocks = blocks hyperparameters = {('a.primitive.Name#2', 'some'): 'arg'} mlpipeline.set_hyperparameters(hyperparameters) block_1.set_hyperparameters.assert_not_called() block_2.set_hyperparameters.assert_called_once_with({'some': 'arg'})
def test_set_hyperparameters(self): block_1 = get_mlblock_mock() block_2 = get_mlblock_mock() blocks = OrderedDict(( ('a.primitive.Name#1', block_1), ('a.primitive.Name#2', block_2), )) mlpipeline = MLPipeline(['a_primitive']) mlpipeline.blocks = blocks hyperparameters = {'a.primitive.Name#2': {'some': 'arg'}} mlpipeline.set_hyperparameters(hyperparameters) block_1.set_hyperparameters.assert_not_called() block_2.set_hyperparameters.assert_called_once_with({'some': 'arg'})