fix:using simplifyOptimizerRules wasn't RemoveEmptyUnionBranches

Author: Max-Cheng

core/trino-main/src/main/java/io/trino/sql/planner/PlanOptimizers.java

@@ -963,6 +963,7 @@ public PlanOptimizers(
                         .add(new PushFilterIntoValues(plannerContext))
                         .add(new ReplaceJoinOverConstantWithProject())
                         .add(new RemoveRedundantPredicateAboveTableScan(plannerContext))
+                        .add(new RemoveEmptyUnionBranches())
                         .build()));
         // Remove unsupported dynamic filters introduced by PredicatePushdown. Also, cleanup dynamic filters removed by
         // PushPredicateIntoTableScan and RemoveRedundantPredicateAboveTableScan due to those rules replacing table scans with empty ValuesNode

core/trino-main/src/test/java/io/trino/sql/planner/TestLogicalPlanner.java

@@ -1332,6 +1332,32 @@ public void testRemovesTrivialFilters()
                         values("nationkey", "name", "regionkey", "comment")));
     }
 
+    @Test
+    public void testRemoveTrivialFiltersWithPartitioningHandle()
+    {
+        assertThat(countOfMatchingNodes(
+                plan("""
+                    SELECT *
+                    FROM (
+                        SELECT n.name AS "col1",
+                          n.nationkey AS "col2",
+                          CAST(null AS varchar) AS "col3"
+                        FROM nation n
+                        WHERE n.nationkey <= 3
+                        UNION ALL
+                        SELECT r.name AS "col1",
+                          CAST(r.regionkey AS bigint) AS "col2",
+                          r.comment AS "col3"
+                        FROM region r
+                      ) subquery
+                    WHERE "col3" IN (
+                        SELECT r.comment
+                        FROM region r
+                      )
+                """),
+                ValuesNode.class::isInstance)).isEqualTo(0);
+    }
+
     @Test
     public void testRemovesNullFilter()
     {