diff --git a/backends-velox/src/test/scala/io/glutenproject/execution/VeloxColumnarCacheSuite.scala b/backends-velox/src/test/scala/io/glutenproject/execution/VeloxColumnarCacheSuite.scala index ac58db67dc66..267a7bc15ce4 100644 --- a/backends-velox/src/test/scala/io/glutenproject/execution/VeloxColumnarCacheSuite.scala +++ b/backends-velox/src/test/scala/io/glutenproject/execution/VeloxColumnarCacheSuite.scala @@ -109,21 +109,23 @@ class VeloxColumnarCacheSuite extends VeloxWholeStageTransformerSuite with Adapt } } - // test("Support transform count(1) with table cache") { - // val cached = spark.table("lineitem").cache() - // try { - // val df = spark.sql("SELECT COUNT(*) FROM lineitem") - // checkAnswer(df, Row(60175)) - // assert( - // find(df.queryExecution.executedPlan) { - // case _: RowToVeloxColumnarExec => true - // case _ => false - // }.isEmpty - // ) - // } finally { - // cached.unpersist() - // } - // } + test("Support transform count(1) with table cache") { + withSQLConf(GlutenConfig.COLUMNAR_TABLE_CACHE_ENABLED.key -> "true") { + val cached = spark.table("lineitem").cache() + try { + val df = spark.sql("SELECT COUNT(*) FROM lineitem") + checkAnswer(df, Row(60175)) + assert( + find(df.queryExecution.executedPlan) { + case _: RowToVeloxColumnarExec => true + case _ => false + }.isEmpty + ) + } finally { + cached.unpersist() + } + } + } test("no ColumnarToRow for table cache") { val cached = spark.table("lineitem").cache()