senthh commented on code in PR #50355:
URL: https://github.com/apache/spark/pull/50355#discussion_r2011235327
##########
sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/TimeExpressionsSuite.scala:
##########
@@ -53,6 +53,61 @@ class TimeExpressionsSuite extends SparkFunSuite with
ExpressionEvalHelper {
parameters = Map("input" -> "'100:50'", "format" -> "'mm:HH'"))
}
+ test("HourExpressionBuilder") {
+ // Empty expressions list
+ checkError(
+ exception = intercept[AnalysisException] {
+ MinuteExpressionBuilder.build("hour", Seq.empty)
+ },
+ condition = "WRONG_NUM_ARGS.WITHOUT_SUGGESTION",
+ parameters = Map(
+ "functionName" -> "`hour`",
+ "expectedNum" -> "> 0",
+ "actualNum" -> "0",
+ "docroot" -> SPARK_DOC_ROOT)
+ )
+
+ // test TIME-typed child should build HoursOfTime
+ val timeExpr = Literal(localTime(12, 58, 59), TimeType())
+ val builtExprForTime = HourExpressionBuilder.build("hour", Seq(timeExpr))
+ assert(builtExprForTime.isInstanceOf[HoursOfTime])
+ assert(builtExprForTime.asInstanceOf[HoursOfTime].child eq timeExpr)
+
+ // test non TIME-typed child should build hour
+ val tsExpr = Literal("2007-09-03 10:45:23")
+ val builtExprForTs = HourExpressionBuilder.build("hour", Seq(tsExpr))
+ assert(builtExprForTs.isInstanceOf[Hour])
+ assert(builtExprForTs.asInstanceOf[Hour].child eq tsExpr)
+ }
+
+ test("Hour with TIME type") {
+ // A few test times in microseconds since midnight:
+ // time in microseconds -> expected minute
+ val testTimes = Seq(
+ localTime() -> 0,
+ localTime(1) -> 1,
+ localTime(0, 59) -> 0,
+ localTime(14, 30) -> 14,
+ localTime(12, 58, 59) -> 12,
+ localTime(23, 0, 1) -> 23,
+ localTime(23, 59, 59, 999999) -> 23
+ )
+
+ // Create a literal with TimeType() for each test microsecond value
+ // evaluate MinutesOfTime(...), and check that the result matches the
expected minute.
Review Comment:
Corrected to HoursOfTime @MaxGekk
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
