Merge branch 'rightlib' into mergelibs-241120-1113

1 files changed, 58 insertions, 0 deletions

diff --git a/yql/essentials/tests/sql/suites/compute_range/in.sql b/yql/essentials/tests/sql/suites/compute_range/in.sql
new file mode 100644
index 0000000000..1c1f3dbee0
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/compute_range/in.sql
@@ -0,0 +1,58 @@
+/* syntax version 1 */
+/* postgres can not */
+/* yt can not */
+
+pragma warning("disable", "4510");
+pragma warning("disable", "1108");
+
+-- basic IN
+select YQL::RangeComputeFor(
+  Struct<x:UInt32>,
+  ($row) -> ($row.x IN (1, 2, -1)),
+  AsTuple(AsAtom("x"))
+);
+
+-- opaque collection
+select YQL::RangeComputeFor(
+  Struct<x:UInt32>,
+  ($row) -> ($row.x IN ListFromRange(-1, 3)),
+  AsTuple(AsAtom("x"))
+);
+
+-- optional collection
+select YQL::RangeComputeFor(
+  Struct<x:UInt32>,
+  ($row) -> (($row.x IN Just(AsSet(-1, 1, 2))) ?? false),
+  AsTuple(AsAtom("x"))
+);
+
+-- optional items
+select YQL::RangeComputeFor(
+  Struct<x:UInt32>,
+  ($row) -> (($row.x IN (-1, 10u, 20, 1/0)) ?? false),
+  AsTuple(AsAtom("x"))
+);
+
+-- tuple
+select YQL::RangeComputeFor(
+  Struct<x:UInt32, y:Uint32, z:Uint32>,
+  ($row) -> (($row.y, $row.x, $row.z) IN [(1,2,3), (100,200,300)]),
+  AsTuple(AsAtom("x"), AsAtom("y"), AsAtom("z"))
+);
+
+
+-- tuple partial
+select YQL::RangeComputeFor(
+  Struct<x:UInt32, y:Uint32, z:Uint32>,
+  ($row) -> (($row.y, $row.x, $row.z) IN [Just(Just((1,2,3))), (100,200,300), null]),
+  AsTuple(AsAtom("x"), AsAtom("y"))
+);
+
+
+-- tuple with implicit nulls
+select YQL::RangeComputeFor(
+  Struct<x:UInt32, y:Uint32, z:Uint32>,
+  ($row) -> (($row.y, $row.x) IN ((1,2,3), (100, 200, 300))),
+  AsTuple(AsAtom("x"), AsAtom("y"))
+);
+