From 6b9ad662f109fa04b67f7113505886b16b5af011 Mon Sep 17 00:00:00 2001 From: Maxim Gekk Date: Thu, 12 Sep 2019 11:39:47 +0500 Subject: [PATCH] Add tests for casting to date --- .../spark/sql/catalyst/expressions/CastSuite.scala | 11 +++++++---- 1 file changed, 7 insertions(+), 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 index 7a781295a7957..590d0024ddaa2 100644 --- 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")) + } }