|
19 | 19 |
|
20 | 20 | package org.apache.comet.exec |
21 | 21 |
|
22 | | -import org.apache.comet.CometSparkSessionExtensions.{isSpark35Plus, isSpark40Plus} |
23 | | -import org.apache.comet.testing.{DataGenOptions, ParquetGenerator, SchemaGenOptions} |
24 | | -import org.apache.comet.{CometConf, ExtendedExplainInfo} |
| 22 | +import java.sql.Date |
| 23 | +import java.time.{Duration, Period} |
| 24 | + |
| 25 | +import scala.util.Random |
| 26 | + |
| 27 | +import org.scalactic.source.Position |
| 28 | +import org.scalatest.Tag |
| 29 | + |
25 | 30 | import org.apache.hadoop.fs.Path |
26 | 31 | import org.apache.spark.sql._ |
| 32 | +import org.apache.spark.sql.catalyst.{FunctionIdentifier, TableIdentifier} |
27 | 33 | import org.apache.spark.sql.catalyst.catalog.{BucketSpec, CatalogStatistics, CatalogTable} |
28 | | -import org.apache.spark.sql.catalyst.expressions.aggregate.{AggregateMode, BloomFilterAggregate} |
29 | 34 | import org.apache.spark.sql.catalyst.expressions.{Expression, ExpressionInfo, Hex} |
30 | | -import org.apache.spark.sql.catalyst.{FunctionIdentifier, TableIdentifier} |
| 35 | +import org.apache.spark.sql.catalyst.expressions.aggregate.{AggregateMode, BloomFilterAggregate} |
31 | 36 | import org.apache.spark.sql.comet._ |
32 | 37 | import org.apache.spark.sql.comet.execution.shuffle.{CometColumnarShuffle, CometShuffleExchangeExec} |
| 38 | +import org.apache.spark.sql.execution.{CollectLimitExec, ProjectExec, SQLExecution, UnionExec} |
33 | 39 | import org.apache.spark.sql.execution.adaptive.{AdaptiveSparkPlanExec, BroadcastQueryStageExec} |
34 | 40 | import org.apache.spark.sql.execution.datasources.parquet.ParquetFileFormat |
35 | 41 | import org.apache.spark.sql.execution.exchange.{BroadcastExchangeExec, ReusedExchangeExec, ShuffleExchangeExec} |
36 | 42 | import org.apache.spark.sql.execution.joins.{BroadcastHashJoinExec, BroadcastNestedLoopJoinExec, CartesianProductExec, SortMergeJoinExec} |
37 | 43 | import org.apache.spark.sql.execution.reuse.ReuseExchangeAndSubquery |
38 | 44 | import org.apache.spark.sql.execution.window.WindowExec |
39 | | -import org.apache.spark.sql.execution.{CollectLimitExec, ProjectExec, SQLExecution, UnionExec} |
40 | 45 | import org.apache.spark.sql.functions._ |
41 | 46 | import org.apache.spark.sql.internal.SQLConf |
42 | 47 | import org.apache.spark.sql.internal.SQLConf.SESSION_LOCAL_TIMEZONE |
43 | 48 | import org.apache.spark.unsafe.types.UTF8String |
44 | | -import org.scalactic.source.Position |
45 | | -import org.scalatest.Tag |
46 | 49 |
|
47 | | -import java.sql.Date |
48 | | -import java.time.{Duration, Period} |
49 | | -import scala.util.Random |
| 50 | +import org.apache.comet.{CometConf, ExtendedExplainInfo} |
| 51 | +import org.apache.comet.CometSparkSessionExtensions.{isSpark35Plus, isSpark40Plus} |
| 52 | +import org.apache.comet.testing.{DataGenOptions, ParquetGenerator, SchemaGenOptions} |
50 | 53 |
|
51 | 54 | class CometExecSuite extends CometTestBase { |
52 | 55 |
|
@@ -331,8 +334,6 @@ class CometExecSuite extends CometTestBase { |
331 | 334 | } |
332 | 335 | } |
333 | 336 |
|
334 | | - |
335 | | - |
336 | 337 | test("fix CometNativeExec.doCanonicalize for ReusedExchangeExec") { |
337 | 338 | withSQLConf( |
338 | 339 | CometConf.COMET_EXEC_BROADCAST_FORCE_ENABLED.key -> "true", |
|
0 commit comments