fix ut

apache · Dec 5, 2023 · 432ad8f · 432ad8f
1 parent 4a6a5a9
commit 432ad8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...rc/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala b/...rc/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala
@@ -196,11 +196,11 @@ class GlutenClickHouseRSSColumnarShuffleAQESuite
       val res = spark.sql("select spark_partition_id(), id from t group by id").collect()
       assert(res.length == 3)
       assert(res(0).getInt(0) == 0)
-      assert(res(0).getInt(1) == 0)
+      assert(res(0).getInt(1) == 2)
       assert(res(1).getInt(0) == 1)
       assert(res(1).getInt(1) == 1)
       assert(res(2).getInt(0) == 2)
-      assert(res(2).getInt(1) == 2)
+      assert(res(2).getInt(1) == 0)
     }
   }
 }