Skip to content

Commit 6cd34d9

Browse files
committed
add tests for arrays
1 parent 9595237 commit 6cd34d9

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/DataFrameSuite.scala

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2280,5 +2280,17 @@ class DataFrameSuite extends QueryTest with SharedSQLContext {
22802280
val h3 = p3.queryExecution.logical.canonicalized.semanticHash()
22812281
assert(h1 == h2)
22822282
assert(h1 != h3)
2283+
2284+
val df2 = Seq(Array(1, 2)).toDF("id")
2285+
val arrays1 = df2.where($"id".isin(lit(Array(1, 2)), lit(Array(2, 1))))
2286+
val arrays2 = df2.where($"id".isin(lit(Array(2, 1)), lit(Array(1, 2))))
2287+
val arrays3 = df2.where($"id".isin(lit(Array(3, 2)), lit(Array(2, 1))))
2288+
assert(arrays1.queryExecution.executedPlan.sameResult(arrays2.queryExecution.executedPlan))
2289+
assert(!arrays1.queryExecution.executedPlan.sameResult(arrays3.queryExecution.executedPlan))
2290+
val arraysHash1 = arrays1.queryExecution.logical.canonicalized.semanticHash()
2291+
val arraysHash2 = arrays2.queryExecution.logical.canonicalized.semanticHash()
2292+
val arraysHash3 = arrays3.queryExecution.logical.canonicalized.semanticHash()
2293+
assert(arraysHash1 == arraysHash2)
2294+
assert(arraysHash1 != arraysHash3)
22832295
}
22842296
}

0 commit comments

Comments
 (0)