Add tests for casting to date

MaxGekk · Sep 12, 2019 · 6b9ad66 · 6b9ad66
1 parent cb1e6c0
commit 6b9ad66
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/CastSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/expressions/CastSuite.scala
@@ -855,10 +855,6 @@ class CastSuite extends SparkFunSuite with ExpressionEvalHelper {
     assert(cast("1234", DateType).checkInputDataTypes().isSuccess)
     assert(cast(new Timestamp(1), DateType).checkInputDataTypes().isSuccess)
     assert(cast(false, DateType).checkInputDataTypes().isFailure)
-    assert(cast(1.toByte, DateType).checkInputDataTypes().isFailure)
-    assert(cast(1.toShort, DateType).checkInputDataTypes().isFailure)
-    assert(cast(1, DateType).checkInputDataTypes().isFailure)
-    assert(cast(1L, DateType).checkInputDataTypes().isFailure)
     assert(cast(1.0.toFloat, DateType).checkInputDataTypes().isFailure)
     assert(cast(1.0, DateType).checkInputDataTypes().isFailure)
   }
@@ -1200,4 +1196,11 @@ class CastSuite extends SparkFunSuite with ExpressionEvalHelper {
       checkEvaluation(cast(Long.MinValue - 0.9D, LongType), Long.MinValue)
     }
   }
+
+  test("cast integral types to date") {
+    checkCast(0.toByte, Date.valueOf("1970-01-01"))
+    checkCast(-1.toShort, Date.valueOf("1969-12-31"))
+    checkCast(1, Date.valueOf("1970-01-02"))
+    checkCast(365, Date.valueOf("1971-01-01"))
+  }
 }