Test for current_date, current_timestamp and now (#3282)

Signed-off-by: Shane Ding <shane200195@gmail.com>
NVIDIA · Aug 25, 2021 · 4cc1d89 · 4cc1d89
1 parent e94afd9
commit 4cc1d89
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/src/test/scala/com/nvidia/spark/rapids/ParseDateTimeSuite.scala b/tests/src/test/scala/com/nvidia/spark/rapids/ParseDateTimeSuite.scala
@@ -357,6 +357,24 @@ class ParseDateTimeSuite extends SparkQueryCompareTestSuite with BeforeAndAfterE
     }
   }
 
+  test("literals: ensure time literals are correct") {
+    val conf = new SparkConf()
+    val df = withGpuSparkSession(spark => {
+      spark.sql("SELECT current_date(), current_timestamp(), now() FROM RANGE(1, 10)")
+    }, conf)
+
+    val times = df.collect()
+    val systemCurrentTime = System.currentTimeMillis()
+    val res = times.forall(time => {
+      val diffDate = systemCurrentTime - time.getDate(0).getTime()
+      val diffTimestamp = systemCurrentTime - time.getTimestamp(1).getTime()
+      val diffNow = systemCurrentTime - time.getTimestamp(2).getTime()
+      diffDate.abs <= 8.64E7 & diffTimestamp.abs <= 1000 & diffNow.abs <= 1000
+    })
+
+    assert(res)
+  }
+
   private def testRegex(rule: RegexReplace, values: Seq[String], expected: Seq[String]): Unit = {
     withResource(ColumnVector.fromStrings(values: _*)) { v =>
       withResource(ColumnVector.fromStrings(expected: _*)) { expected =>