Skip to content

Commit deec886

Browse files
committed
[AURON apache#1654] Add AuronQueryTest base class for SQL testing (apache#1655)
* Introduce AuronQueryTest * fixup * fixup * fixup * fixup * fixup * fixup * fixup * fixup * fixup * fixup * format code style
1 parent 87e688d commit deec886

13 files changed

Lines changed: 258 additions & 373 deletions

spark-extension-shims-spark/src/test/scala/org/apache/spark/sql/auron/AuronCheckConvertBroadcastExchangeSuite.scala renamed to spark-extension-shims-spark/src/test/scala/org.apache.auron/AuronCheckConvertBroadcastExchangeSuite.scala

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,19 +14,19 @@
1414
* See the License for the specific language governing permissions and
1515
* limitations under the License.
1616
*/
17-
package org.apache.spark.sql.auron
17+
package org.apache.auron
1818

19-
import org.apache.spark.sql.{QueryTest, Row, SparkSession}
19+
import org.apache.spark.sql.{AuronQueryTest, Row, SparkSession}
20+
import org.apache.spark.sql.auron.AuronConverters
2021
import org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanExec
2122
import org.apache.spark.sql.execution.auron.plan.NativeBroadcastExchangeExec
2223
import org.apache.spark.sql.execution.exchange.BroadcastExchangeExec
2324
import org.apache.spark.sql.test.SharedSparkSession
2425

2526
class AuronCheckConvertBroadcastExchangeSuite
26-
extends QueryTest
27+
extends AuronQueryTest
2728
with SharedSparkSession
28-
with AuronSQLTestHelper
29-
with org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanHelper {
29+
with AuronSQLTestHelper {
3030
import testImplicits._
3131

3232
test(

spark-extension-shims-spark/src/test/scala/org/apache/spark/sql/auron/AuronCheckConvertShuffleExchangeSuite.scala renamed to spark-extension-shims-spark/src/test/scala/org.apache.auron/AuronCheckConvertShuffleExchangeSuite.scala

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,18 +14,18 @@
1414
* See the License for the specific language governing permissions and
1515
* limitations under the License.
1616
*/
17-
package org.apache.spark.sql.auron
17+
package org.apache.auron
1818

19-
import org.apache.spark.sql.{QueryTest, Row, SparkSession}
19+
import org.apache.spark.sql.{AuronQueryTest, Row, SparkSession}
20+
import org.apache.spark.sql.auron.AuronConverters
2021
import org.apache.spark.sql.execution.auron.plan.NativeShuffleExchangeExec
2122
import org.apache.spark.sql.execution.exchange.ShuffleExchangeExec
2223
import org.apache.spark.sql.test.SharedSparkSession
2324

2425
class AuronCheckConvertShuffleExchangeSuite
25-
extends QueryTest
26+
extends AuronQueryTest
2627
with SharedSparkSession
27-
with AuronSQLTestHelper
28-
with org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanHelper {
28+
with AuronSQLTestHelper {
2929

3030
test(
3131
"test set auron shuffle manager convert to native shuffle exchange where set spark.auron.enable is true") {

spark-extension-shims-spark/src/test/scala/org/apache/spark/sql/auron/AuronEmptyNativeRddSuite.scala renamed to spark-extension-shims-spark/src/test/scala/org.apache.auron/AuronEmptyNativeRddSuite.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,15 @@
1414
* See the License for the specific language governing permissions and
1515
* limitations under the License.
1616
*/
17-
package org.apache.spark.sql.auron
17+
package org.apache.auron
1818

19+
import org.apache.spark.sql.AuronQueryTest
20+
import org.apache.spark.sql.auron.{AuronConverters, EmptyNativeRDD, NativeRDD}
1921
import org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanExec
2022
import org.apache.spark.sql.execution.auron.plan.{NativeOrcScanExec, NativeParquetScanExec}
2123

2224
class AuronEmptyNativeRddSuite
23-
extends org.apache.spark.sql.QueryTest
25+
extends AuronQueryTest
2426
with BaseAuronSQLSuite
2527
with AuronSQLTestHelper {
2628

0 commit comments

Comments
 (0)