Skip to content

Commit bbd12bf

Browse files
gstvggstvg
committed
tests: add sqllogictests using captured columns
1 parent b61236f commit bbd12bf

1 file changed

Lines changed: 54 additions & 24 deletions

File tree

‎datafusion/sqllogictest/test_files/multi_arg_lambda.slt‎

Lines changed: 54 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -18,45 +18,75 @@
1818
statement ok
1919
set datafusion.sql_parser.dialect = databricks;
2020

21-
query I
22-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+1);
21+
statement ok
22+
CREATE TABLE t AS SELECT 1 as x;
23+
24+
query II
25+
SELECT
26+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+1),
27+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+x)
28+
from t;
2329
----
24-
2
30+
2 2
2531

26-
query I
27-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+1);
32+
query II
33+
SELECT
34+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+1),
35+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+x)
36+
from t;
2837
----
29-
3
38+
3 3
3039

31-
query I
32-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> c+1);
40+
query II
41+
SELECT
42+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> c+1),
43+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> c+x)
44+
from t;
3345
----
34-
4
46+
4 4
3547

36-
query I
37-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b);
48+
query II
49+
SELECT
50+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b+1),
51+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b+x)
52+
from t;
3853
----
39-
3
54+
4 4
4055

41-
query I
42-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+c);
56+
query II
57+
SELECT
58+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+c+1),
59+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+c+x)
60+
from t;
4361
----
44-
4
62+
5 5
4563

46-
query I
47-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+c);
64+
query II
65+
SELECT
66+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+c+1),
67+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> b+c+x)
68+
from t;
4869
----
49-
5
70+
6 6
5071

51-
query I
52-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b+c);
72+
query II
73+
SELECT
74+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b+c+1),
75+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> a+b+c+x)
76+
from t;
5377
----
54-
6
78+
7 7
5579

56-
query I
57-
SELECT multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> 1);
80+
query II
81+
SELECT
82+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> 1),
83+
multi_arg_higher_order_udf(1, 2, 3, (a, b, c) -> x)
84+
from t;
5885
----
59-
1
86+
1 1
6087

6188
statement ok
6289
set datafusion.sql_parser.dialect = generic;
90+
91+
statement ok
92+
DROP TABLE t;

0 commit comments

Comments
 (0)