diff --git a/velox/exec/tests/HashJoinTest.cpp b/velox/exec/tests/HashJoinTest.cpp index 3ad02678d3b6..c32f33fe2226 100644 --- a/velox/exec/tests/HashJoinTest.cpp +++ b/velox/exec/tests/HashJoinTest.cpp @@ -8254,7 +8254,8 @@ TEST_F(HashJoinTest, combineSmallVectorsAfterFilter) { return makeRowVector( {"t0", "t1"}, { - makeFlatVector(1'000, [](auto row) { return row; }), + makeFlatVector( + 1'000, [](auto row) { return row; }, nullEvery(400)), makeFlatVector(1'000, [](auto row) { return row * 10; }), }); }); @@ -8264,7 +8265,9 @@ TEST_F(HashJoinTest, combineSmallVectorsAfterFilter) { {"u0", "u1"}, { makeFlatVector( - 1'000, [](auto row) { return -100 + (row / 5); }), + 1'000, + [](auto row) { return -100 + (row / 5); }, + nullEvery(300)), makeFlatVector( 1'000, [](auto row) { return -1000 + (row / 5) * 10; }), }); @@ -8365,7 +8368,7 @@ TEST_F(HashJoinTest, combineSmallVectorsAfterFilter) { verifyJoinOutputVectorCount( 1, // 3 output vectors are merged to 1 vector. core::JoinType::kLeftSemiProject, - "SELECT t0, t1, t0 IN (SELECT u0 FROM u WHERE (t1 + u1) % 3 = 0) FROM t"); + "SELECT t0, t1, EXISTS (SELECT u0 FROM u WHERE t0 = u0 AND (t1 + u1) % 3 = 0) FROM t"); verifyJoinOutputVectorCount( 1, // 3 output vectors are merged to 1 vector. core::JoinType::kLeftSemiProject, @@ -8374,7 +8377,7 @@ TEST_F(HashJoinTest, combineSmallVectorsAfterFilter) { verifyJoinOutputVectorCount( 1, // 3 output vectors are merged to 1 vector. core::JoinType::kLeftSemiProject, // Flip join side. - "SELECT t0, t1, t0 IN (SELECT u0 FROM u WHERE (t1 + u1) % 3 = 0) FROM t", + "SELECT t0, t1, EXISTS (SELECT u0 FROM u WHERE t0 = u0 AND (t1 + u1) % 3 = 0) FROM t", false, true); } @@ -8396,11 +8399,11 @@ TEST_F(HashJoinTest, combineSmallVectorsAfterFilter) { verifyJoinOutputVectorCount( 1, // 3 output vectors are merged to 1 vector. core::JoinType::kAnti, - "SELECT t0, t1, FROM t WHERE NOT EXISTS (SELECT * FROM u WHERE t0 = u0 AND (t1 + u1) % 3 = 0)"); + "SELECT t0, t1 FROM t WHERE NOT EXISTS (SELECT * FROM u WHERE t0 = u0 AND (t1 + u1) % 3 = 0)"); verifyJoinOutputVectorCount( 1, // 2 output vectors are merged to 1 vector. core::JoinType::kAnti, - "SELECT t0, t1, FROM t WHERE NOT EXISTS (SELECT * FROM u WHERE t0 = u0 AND (t1 + u1) % 3 = 0)", + "SELECT t0, t1 FROM t WHERE t0 NOT IN (SELECT u0 FROM u WHERE (t1 + u1) % 3 = 0)", true); } }