File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
sql/core/src/test/scala/org/apache/spark/sql/execution/datasources Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff 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`
You can’t perform that action at this time.
0 commit comments