Skip to content

Commit e555a1c

Browse files
committed
Add one test.
1 parent 6feaaa4 commit e555a1c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

sql/core/src/test/scala/org/apache/spark/sql/execution/datasources/DataSourceStrategySuite.scala

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -290,9 +290,21 @@ class DataSourceStrategySuite extends PlanTest with SharedSparkSession {
290290
"an expression that can be pushed down") {
291291
attrInts.foreach { case (attrInt, colName) =>
292292
assert(PushableColumnAndNestedColumn.unapply(attrInt) === Some(colName))
293+
294+
if (colName.contains(".")) {
295+
assert(PushableColumnWithoutNestedColumn.unapply(attrInt) === None)
296+
} else {
297+
assert(PushableColumnWithoutNestedColumn.unapply(attrInt) === Some(colName))
298+
}
293299
}
294300
attrStrs.foreach { case (attrStr, colName) =>
295301
assert(PushableColumnAndNestedColumn.unapply(attrStr) === Some(colName))
302+
303+
if (colName.contains(".")) {
304+
assert(PushableColumnWithoutNestedColumn.unapply(attrStr) === None)
305+
} else {
306+
assert(PushableColumnWithoutNestedColumn.unapply(attrStr) === Some(colName))
307+
}
296308
}
297309

298310
// `Abs(col)` can not be pushed down, so it returns `None`

0 commit comments

Comments
 (0)