Skip to content

Commit 60bceb5

Browse files
committed
test: add column-based nested array coverage for array_position
1 parent 7558c66 commit 60bceb5

1 file changed

Lines changed: 29 additions & 0 deletions

File tree

spark/src/test/resources/sql-tests/expressions/array/array_position.sql

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,35 @@ SELECT array_position(array(array(1, 2), array(3, 4)), array(1, 2))
225225
query spark_answer_only
226226
SELECT array_position(array(array(1, 2), array(3, 4)), array(5, 6))
227227

228+
-- nested int array column (exercises position_fallback natively)
229+
statement
230+
CREATE TABLE test_ap_nested_int(arr array<array<int>>, val array<int>) USING parquet
231+
232+
statement
233+
INSERT INTO test_ap_nested_int VALUES
234+
(array(array(1, 2), array(3, 4), array(5, 6)), array(3, 4)),
235+
(array(array(1, 2), array(3, 4)), array(5, 6)),
236+
(array(array(1, 2), array(1, 2, 3)), array(1, 2)),
237+
(NULL, array(1, 2)),
238+
(array(array(1, 2)), NULL)
239+
240+
query
241+
SELECT array_position(arr, val) FROM test_ap_nested_int
242+
243+
-- nested string array column (exercises position_fallback natively)
244+
statement
245+
CREATE TABLE test_ap_nested_str(arr array<array<string>>, val array<string>) USING parquet
246+
247+
statement
248+
INSERT INTO test_ap_nested_str VALUES
249+
(array(array('a', 'b'), array('c', 'd')), array('c', 'd')),
250+
(array(array('a', 'b'), array('c', 'd')), array('x', 'y')),
251+
(NULL, array('a')),
252+
(array(array('a')), NULL)
253+
254+
query
255+
SELECT array_position(arr, val) FROM test_ap_nested_str
256+
228257
-- timestamp arrays
229258
statement
230259
CREATE TABLE test_ap_ts(arr array<timestamp>, val timestamp) USING parquet

0 commit comments

Comments
 (0)