Skip to content

Commit d3925bc

Browse files
committed
Update ExpressionEvalHelper.checkResult to properly compare different InternalRow implementations
1 parent 426d856 commit d3925bc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/ExpressionEvalHelper.scala

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,10 @@ trait ExpressionEvalHelper extends GeneratorDrivenPropertyChecks with PlanTestBa
138138
case (result: Float, expected: Float) =>
139139
if (expected.isNaN) result.isNaN else expected == result
140140
case (result: Row, expected: InternalRow) => result.toSeq == expected.toSeq(result.schema)
141+
case (result: Seq[InternalRow], expected: Seq[InternalRow]) =>
142+
result.size == expected.size && result.zip(expected).forall { case (r, e) =>
143+
checkResult(r, e, exprDataType, exprNullable)
144+
}
141145
case _ =>
142146
result == expected
143147
}

0 commit comments

Comments
 (0)