Skip to content

Commit d4dc9ca

Browse files
author
Andrew Or
committed
Fix tests
1 parent 94ca5de commit d4dc9ca

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

sql/core/src/test/scala/org/apache/spark/sql/execution/aggregate/TungstenAggregationIteratorSuite.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ package org.apache.spark.sql.execution.aggregate
2020
import org.apache.spark._
2121
import org.apache.spark.sql.catalyst.expressions._
2222
import org.apache.spark.sql.catalyst.expressions.InterpretedMutableProjection
23+
import org.apache.spark.sql.execution.metric.LongSQLMetric
2324
import org.apache.spark.sql.test.TestSQLContext
2425
import org.apache.spark.unsafe.memory.TaskMemoryManager
2526

@@ -39,8 +40,9 @@ class TungstenAggregationIteratorSuite extends SparkFunSuite {
3940
val newMutableProjection = (expr: Seq[Expression], schema: Seq[Attribute]) => {
4041
() => new InterpretedMutableProjection(expr, schema)
4142
}
42-
iter = new TungstenAggregationIterator(
43-
Seq.empty, Seq.empty, Seq.empty, 0, Seq.empty, newMutableProjection, Seq.empty, None)
43+
val dummyAccum = new LongSQLMetric("dummy")
44+
iter = new TungstenAggregationIterator(Seq.empty, Seq.empty, Seq.empty, 0,
45+
Seq.empty, newMutableProjection, Seq.empty, None, dummyAccum, dummyAccum)
4446
val numPages = iter.getHashMap.getNumDataPages
4547
assert(numPages === 1)
4648
} finally {

0 commit comments

Comments
 (0)