From 432ad8f678410d6c1968aee51d3107cec578c537 Mon Sep 17 00:00:00 2001 From: Zhichao Zhang Date: Tue, 5 Dec 2023 16:28:39 +0800 Subject: [PATCH] fix ut --- .../GlutenClickHouseRSSColumnarShuffleAQESuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala b/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala index 5b283299ece55..99a64d7fb4f7c 100644 --- a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala +++ b/gluten-celeborn/clickhouse/src/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) } } }