use temp view

linhongliu-db · linhongliu-db · commit 737f9965ca6e · 2020-09-10T10:55:44.000+08:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala
@@ -2557,12 +2557,14 @@ class DataFrameSuite extends QueryTest
   }
 
   test("SPARK-32816: aggregating multiple distinct DECIMAL columns") {
-    spark.range(0, 100, 1, 1)
-      .selectExpr("id", "cast(id as decimal(9, 0)) as decimal_col")
-      .createOrReplaceTempView("test_table")
-    checkAnswer(
-      sql("select avg(distinct decimal_col), sum(distinct decimal_col) from test_table"),
-      Row(49.5, 4950))
+    withTempView("test_table") {
+      spark.range(0, 100, 1, 1)
+        .selectExpr("id", "cast(id as decimal(9, 0)) as decimal_col")
+        .createOrReplaceTempView("test_table")
+      checkAnswer(
+        sql("select avg(distinct decimal_col), sum(distinct decimal_col) from test_table"),
+        Row(49.5, 4950))
+    }
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -2557,12 +2557,14 @@ class DataFrameSuite extends QueryTest`
`2557`	`2557`	`}`
`2558`	`2558`
`2559`	`2559`	`test("SPARK-32816: aggregating multiple distinct DECIMAL columns") {`
`2560`		`- spark.range(0, 100, 1, 1)`
`2561`		`- .selectExpr("id", "cast(id as decimal(9, 0)) as decimal_col")`
`2562`		`- .createOrReplaceTempView("test_table")`
`2563`		`- checkAnswer(`
`2564`		`- sql("select avg(distinct decimal_col), sum(distinct decimal_col) from test_table"),`
`2565`		`- Row(49.5, 4950))`
	`2560`	`+ withTempView("test_table") {`
	`2561`	`+ spark.range(0, 100, 1, 1)`
	`2562`	`+ .selectExpr("id", "cast(id as decimal(9, 0)) as decimal_col")`
	`2563`	`+ .createOrReplaceTempView("test_table")`
	`2564`	`+ checkAnswer(`
	`2565`	`+ sql("select avg(distinct decimal_col), sum(distinct decimal_col) from test_table"),`
	`2566`	`+ Row(49.5, 4950))`
	`2567`	`+ }`
`2566`	`2568`	`}`
`2567`	`2569`	`}`
`2568`	`2570`