Skip to content

Commit 2f643d4

Browse files
committed
Take care of 1 row case.
1 parent 832db06 commit 2f643d4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/aggregate/Covariance.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ case class CovSample(
165165
cov
166166
}
167167
} else {
168-
0.0
168+
null
169169
}
170170
} else {
171171
null

sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/AggregationQuerySuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -832,8 +832,8 @@ abstract class AggregationQuerySuite extends QueryTest with SQLTestUtils with Te
832832

833833
// one row test
834834
val df3 = Seq.tabulate(1)(x => (1 * x, x * x * x - 2)).toDF("a", "b")
835-
val cov_samp3 = df3.groupBy().agg(covar_samp("a", "b")).collect()(0).getDouble(0)
836-
assert(cov_samp3 == 0.0)
835+
val cov_samp3 = df3.groupBy().agg(covar_samp("a", "b")).collect()(0).get(0)
836+
assert(cov_samp3 == null)
837837

838838
val cov_pop3 = df3.groupBy().agg(covar_pop("a", "b")).collect()(0).getDouble(0)
839839
assert(cov_pop3 == 0.0)

0 commit comments

Comments
 (0)