day and year interval test cases are also support

apache · Jun 13, 2020 · c1765e1 · c1765e1
1 parent f970add
commit c1765e1
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/...src/test/scala/org/apache/spark/sql/catalyst/expressions/CollectionExpressionsSuite.scala b/...src/test/scala/org/apache/spark/sql/catalyst/expressions/CollectionExpressionsSuite.scala
@@ -1838,11 +1838,20 @@ class CollectionExpressionsSuite extends SparkFunSuite with ExpressionEvalHelper
   }
 
   test("SPARK-31980: Start and end equal in month range") {
+    checkEvaluation(new Sequence(
+      Literal(Date.valueOf("2018-01-01")),
+      Literal(Date.valueOf("2018-01-01")),
+      Literal(stringToInterval("interval 1 day"))),
+      Seq(Date.valueOf("2018-01-01")))
     checkEvaluation(new Sequence(
       Literal(Date.valueOf("2018-01-01")),
       Literal(Date.valueOf("2018-01-01")),
       Literal(stringToInterval("interval 1 month"))),
-      Seq(
-        Date.valueOf("2018-01-01")))
+      Seq(Date.valueOf("2018-01-01")))
+    checkEvaluation(new Sequence(
+      Literal(Date.valueOf("2018-01-01")),
+      Literal(Date.valueOf("2018-01-01")),
+      Literal(stringToInterval("interval 1 year"))),
+      Seq(Date.valueOf("2018-01-01")))
   }
 }