def test_visit_create_udf(self, mock, l_create_udf_mock):
convertor = StatementToPlanConvertor()
stmt = MagicMock()
stmt.name = 'name'
stmt.if_not_exists = True
stmt.inputs = ['inp']
stmt.outputs = ['out']
stmt.impl_path = 'tmp.py'
stmt.udf_type = 'classification'
mock.side_effect = ['inp', 'out']
convertor.visit_create_udf(stmt)
mock.assert_any_call(stmt.inputs, True)
mock.assert_any_call(stmt.outputs, False)
l_create_udf_mock.assert_called_once()
l_create_udf_mock.assert_called_with(stmt.name, stmt.if_not_exists,
'inp', 'out', stmt.impl_path,
stmt.udf_type)
def test_visit_should_call_create_udf(self):
stmt = MagicMock(spec=CreateUDFStatement)
convertor = StatementToPlanConvertor()
mock = MagicMock()
convertor.visit_create_udf = mock
convertor.visit(stmt)
mock.assert_called_once()
mock.assert_called_with(stmt)
