fsk119 commented on code in PR #27106:
URL: https://github.com/apache/flink/pull/27106#discussion_r2427794148
##########
flink-python/pyflink/table/tests/test_udf.py:
##########
@@ -697,6 +697,192 @@ def decimal_cut_func(decimal_param):
"{1=flink, 2=pyflink},
1000000000000000000.050000000000000000, "
"1000000000000000000.059999999999999999]"])
+ def test_using_udf_in_sql(self):
+ sql = f"""
+ CREATE TEMPORARY FUNCTION echo AS
+ '{UserDefinedFunctionTests.__module__}.echo'
+ LANGUAGE PYTHON
+ """
+ self.t_env.execute_sql(sql)
+ self.assert_equals(
+ list(self.t_env.execute_sql("SELECT echo(1)").collect()),
+ [Row(1)])
+
+ def test_all_data_types_string(self):
+ @udf(result_type='BOOLEAN')
+ def boolean_func(bool_param):
+ assert isinstance(bool_param, bool), 'bool_param of wrong type %s
!' \
+ % type(bool_param)
+ return bool_param
+
+ @udf(result_type='TINYINT')
+ def tinyint_func(tinyint_param):
+ assert isinstance(tinyint_param, int), 'tinyint_param of wrong
type %s !' \
+ % type(tinyint_param)
+ return tinyint_param
+
+ @udf(result_type='SMALLINT')
+ def smallint_func(smallint_param):
+ assert isinstance(smallint_param, int), 'smallint_param of wrong
type %s !' \
+ % type(smallint_param)
+ assert smallint_param == 32767, 'smallint_param of wrong value %s'
% smallint_param
+ return smallint_param
+
+ @udf(result_type='INT')
+ def int_func(int_param):
+ assert isinstance(int_param, int), 'int_param of wrong type %s !' \
+ % type(int_param)
+ assert int_param == -2147483648, 'int_param of wrong value %s' %
int_param
+ return int_param
+
+ @udf(result_type='BIGINT')
+ def bigint_func(bigint_param):
+ assert isinstance(bigint_param, int), 'bigint_param of wrong type
%s !' \
+ % type(bigint_param)
+ return bigint_param
+
+ @udf(result_type='BIGINT')
+ def bigint_func_none(bigint_param):
+ assert bigint_param is None, 'bigint_param %s should be None!' %
bigint_param
+ return bigint_param
+
+ @udf(result_type='FLOAT')
+ def float_func(float_param):
+ assert isinstance(float_param, float) and float_equal(float_param,
1.23, 1e-6), \
+ 'float_param is wrong value %s !' % float_param
+ return float_param
+
+ @udf(result_type='DOUBLE')
+ def double_func(double_param):
+ assert isinstance(double_param, float) and
float_equal(double_param, 1.98932, 1e-7), \
+ 'double_param is wrong value %s !' % double_param
+ return double_param
+
+ @udf(result_type='BYTES')
+ def bytes_func(bytes_param):
+ assert bytes_param == b'flink', \
+ 'bytes_param is wrong value %s !' % bytes_param
+ return bytes_param
+
+ @udf(result_type='STRING')
+ def str_func(str_param):
+ assert str_param == 'pyflink', \
+ 'str_param is wrong value %s !' % str_param
+ return str_param
+
+ @udf(result_type='DATE')
+ def date_func(date_param):
+ from datetime import date
+ assert date_param == date(year=2014, month=9, day=13), \
+ 'date_param is wrong value %s !' % date_param
+ return date_param
+
+ @udf(result_type='TIME')
+ def time_func(time_param):
+ from datetime import time
+ assert time_param == time(hour=12, minute=0, second=0,
microsecond=123000), \
+ 'time_param is wrong value %s !' % time_param
+ return time_param
+
+ @udf(result_type='TIMESTAMP(3)')
Review Comment:
I test timestamp_ltz and I find the results is not correct. I will open
another issue to track this.
##########
flink-python/pyflink/table/tests/test_udf.py:
##########
@@ -697,6 +697,192 @@ def decimal_cut_func(decimal_param):
"{1=flink, 2=pyflink},
1000000000000000000.050000000000000000, "
"1000000000000000000.059999999999999999]"])
+ def test_using_udf_in_sql(self):
+ sql = f"""
+ CREATE TEMPORARY FUNCTION echo AS
+ '{UserDefinedFunctionTests.__module__}.echo'
+ LANGUAGE PYTHON
+ """
+ self.t_env.execute_sql(sql)
+ self.assert_equals(
+ list(self.t_env.execute_sql("SELECT echo(1)").collect()),
+ [Row(1)])
+
+ def test_all_data_types_string(self):
+ @udf(result_type='BOOLEAN')
+ def boolean_func(bool_param):
+ assert isinstance(bool_param, bool), 'bool_param of wrong type %s
!' \
+ % type(bool_param)
+ return bool_param
+
+ @udf(result_type='TINYINT')
+ def tinyint_func(tinyint_param):
+ assert isinstance(tinyint_param, int), 'tinyint_param of wrong
type %s !' \
+ % type(tinyint_param)
+ return tinyint_param
+
+ @udf(result_type='SMALLINT')
+ def smallint_func(smallint_param):
+ assert isinstance(smallint_param, int), 'smallint_param of wrong
type %s !' \
+ % type(smallint_param)
+ assert smallint_param == 32767, 'smallint_param of wrong value %s'
% smallint_param
+ return smallint_param
+
+ @udf(result_type='INT')
+ def int_func(int_param):
+ assert isinstance(int_param, int), 'int_param of wrong type %s !' \
+ % type(int_param)
+ assert int_param == -2147483648, 'int_param of wrong value %s' %
int_param
+ return int_param
+
+ @udf(result_type='BIGINT')
+ def bigint_func(bigint_param):
+ assert isinstance(bigint_param, int), 'bigint_param of wrong type
%s !' \
+ % type(bigint_param)
+ return bigint_param
+
+ @udf(result_type='BIGINT')
+ def bigint_func_none(bigint_param):
+ assert bigint_param is None, 'bigint_param %s should be None!' %
bigint_param
+ return bigint_param
+
+ @udf(result_type='FLOAT')
+ def float_func(float_param):
+ assert isinstance(float_param, float) and float_equal(float_param,
1.23, 1e-6), \
+ 'float_param is wrong value %s !' % float_param
+ return float_param
+
+ @udf(result_type='DOUBLE')
+ def double_func(double_param):
+ assert isinstance(double_param, float) and
float_equal(double_param, 1.98932, 1e-7), \
+ 'double_param is wrong value %s !' % double_param
+ return double_param
+
+ @udf(result_type='BYTES')
+ def bytes_func(bytes_param):
+ assert bytes_param == b'flink', \
+ 'bytes_param is wrong value %s !' % bytes_param
+ return bytes_param
+
+ @udf(result_type='STRING')
+ def str_func(str_param):
+ assert str_param == 'pyflink', \
+ 'str_param is wrong value %s !' % str_param
+ return str_param
+
+ @udf(result_type='DATE')
+ def date_func(date_param):
+ from datetime import date
+ assert date_param == date(year=2014, month=9, day=13), \
+ 'date_param is wrong value %s !' % date_param
+ return date_param
+
+ @udf(result_type='TIME')
+ def time_func(time_param):
+ from datetime import time
+ assert time_param == time(hour=12, minute=0, second=0,
microsecond=123000), \
+ 'time_param is wrong value %s !' % time_param
+ return time_param
+
+ @udf(result_type='TIMESTAMP(3)')
+ def timestamp_func(timestamp_param):
+ from datetime import datetime
+ assert timestamp_param == datetime(2018, 3, 11, 3, 0, 0, 123000), \
+ 'timestamp_param is wrong value %s !' % timestamp_param
+ return timestamp_param
+
+ @udf(result_type='ARRAY<BIGINT>')
+ def array_func(array_param):
+ assert array_param == [[1, 2, 3]] or array_param == ((1, 2, 3),), \
+ 'array_param is wrong value %s !' % array_param
+ return array_param[0]
+
+ @udf(result_type='MAP<BIGINT, STRING>')
Review Comment:
Added
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
