def test__get_outputs_defaults(self):
self_ = MagicMock(autospec=MLPipeline)
pipeline = dict()
outputs = {'default': ['some', 'outputs']}
returned = MLPipeline._get_outputs(self_, pipeline, outputs)
expected = {'default': ['some', 'outputs']}
assert returned == expected
self_._get_block_outputs.assert_not_called()
def test__get_outputs_no_outputs(self):
self_ = MagicMock(autospec=MLPipeline)
self_._last_block_name = 'last_block'
self_._get_block_outputs.return_value = ['some', 'outputs']
pipeline = dict()
outputs = None
returned = MLPipeline._get_outputs(self_, pipeline, outputs)
expected = {'default': ['some', 'outputs']}
assert returned == expected
self_._get_block_outputs.assert_called_once_with('last_block')
def test__get_outputs_additional(self):
self_ = Mock()
pipeline = {
'outputs': {
'default': ['some', 'outputs'],
'additional': ['other', 'outputs']
}
}
outputs = None
returned = MLPipeline._get_outputs(self_, pipeline, outputs)
expected = {
'default': ['some', 'outputs'],
'additional': ['other', 'outputs']
}
assert returned == expected
self_._get_block_outputs.assert_not_called()
