Skip to content

test: cover float IN list predicates #23373

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
alamb merged 1 commit into from
Jul 7, 2026
+58 −0
Merged

test: cover float IN list predicates #23373

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 58 additions & 0 deletions datafusion/sqllogictest/test_files/in_list.slt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -236,3 +236,61 @@ nulls NULL NULL NULL NULL NULL NULL NULL NULL
# Cleanup
statement ok
DROP TABLE in_list_ints_nullable

####
## Float IN List Specializations
####

statement ok
CREATE TABLE in_list_floats AS
SELECT
label,
arrow_cast(value, 'Float16') AS f16,
arrow_cast(value, 'Float32') AS f32,
arrow_cast(value, 'Float64') AS f64
FROM (VALUES
('match', 11.0),
('no_match', 7.0),
('nulls', NULL)
) AS t(label, value);

# Five element IN lists cover the specialized Float16/32/64 paths.
query TTBBB
SELECT
'Float16',
label,
f16 IN (arrow_cast(3.0, 'Float16'), arrow_cast(4.0, 'Float16'), arrow_cast(5.0, 'Float16'), arrow_cast(6.0, 'Float16'), arrow_cast(11.0, 'Float16')),
f16 IN (arrow_cast(NULL, 'Float16'), arrow_cast(3.0, 'Float16'), arrow_cast(4.0, 'Float16'), arrow_cast(5.0, 'Float16'), arrow_cast(11.0, 'Float16')),
f16 IN (arrow_cast(3.0, 'Float16'), arrow_cast(4.0, 'Float16'), arrow_cast(5.0, 'Float16'), arrow_cast(6.0, 'Float16'), arrow_cast(8.0, 'Float16'))
FROM in_list_floats
UNION ALL
SELECT
'Float32',
label,
f32 IN (arrow_cast(3.0, 'Float32'), arrow_cast(4.0, 'Float32'), arrow_cast(5.0, 'Float32'), arrow_cast(6.0, 'Float32'), arrow_cast(11.0, 'Float32')),
f32 IN (arrow_cast(NULL, 'Float32'), arrow_cast(3.0, 'Float32'), arrow_cast(4.0, 'Float32'), arrow_cast(5.0, 'Float32'), arrow_cast(11.0, 'Float32')),
f32 IN (arrow_cast(3.0, 'Float32'), arrow_cast(4.0, 'Float32'), arrow_cast(5.0, 'Float32'), arrow_cast(6.0, 'Float32'), arrow_cast(8.0, 'Float32'))
FROM in_list_floats
UNION ALL
SELECT
'Float64',
label,
f64 IN (arrow_cast(3.0, 'Float64'), arrow_cast(4.0, 'Float64'), arrow_cast(5.0, 'Float64'), arrow_cast(6.0, 'Float64'), arrow_cast(11.0, 'Float64')),
f64 IN (arrow_cast(NULL, 'Float64'), arrow_cast(3.0, 'Float64'), arrow_cast(4.0, 'Float64'), arrow_cast(5.0, 'Float64'), arrow_cast(11.0, 'Float64')),
f64 IN (arrow_cast(3.0, 'Float64'), arrow_cast(4.0, 'Float64'), arrow_cast(5.0, 'Float64'), arrow_cast(6.0, 'Float64'), arrow_cast(8.0, 'Float64'))
FROM in_list_floats
ORDER BY 1, 2
----
Float16 match true true false
Float16 no_match false NULL false
Float16 nulls NULL NULL NULL
Float32 match true true false
Float32 no_match false NULL false
Float32 nulls NULL NULL NULL
Float64 match true true false
Float64 no_match false NULL false
Float64 nulls NULL NULL NULL

# Cleanup
statement ok
DROP TABLE in_list_floats
Loading