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