Skip to content
Prev Previous commit
Next Next commit
Add some more cases for exceptions
  • Loading branch information
HyukjinKwon committed Nov 4, 2016
commit 7bc0a192c3887cbd8806dae9eeb5f73fe731b80a
10 changes: 10 additions & 0 deletions sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala
Original file line number Diff line number Diff line change
Expand Up @@ -1741,5 +1741,15 @@ class DataFrameSuite extends QueryTest with SharedSQLContext {
testData.selectExpr("randn(cast(NULL AS LONG))"), testData.selectExpr("randn(0L)"))
checkAnswer(
testData.selectExpr("randn(cast(3L / 12L AS LONG))"), testData.selectExpr("randn(0L)"))

val eOne = intercept[AnalysisException] {
testData.selectExpr("rand(key)").collect()
}
assert(eOne.message.contains("Input argument to rand must be an integer/long literal."))

val eTwo = intercept[AnalysisException] {
testData.selectExpr("randn(key)").collect()
}
assert(eTwo.message.contains("Input argument to randn must be an integer/long literal."))
}
}