Add tests for to timestamp function

jpedroantunes · jpedroantunes · commit f09022011a69 · 2021-05-17T08:21:21.000-03:00
diff --git a/cpp/src/gandiva/precompiled/time_test.cc b/cpp/src/gandiva/precompiled/time_test.cc
@@ -748,48 +748,68 @@ TEST(TestTime, TestToTimestamp) {
   auto seconds = 0;
   EXPECT_EQ(ts, to_timestamp_int32(seconds));
   EXPECT_EQ(ts, to_timestamp_int64(seconds));
+  EXPECT_EQ(ts, to_timestamp_float32(seconds));
+  EXPECT_EQ(ts, to_timestamp_float64(seconds));
 
   ts = StringToTimestamp("1970-01-01 00:00:01");
   seconds = 1;
   EXPECT_EQ(ts, to_timestamp_int32(seconds));
   EXPECT_EQ(ts, to_timestamp_int64(seconds));
+  EXPECT_EQ(ts, to_timestamp_float32(seconds));
+  EXPECT_EQ(ts, to_timestamp_float64(seconds));
 
   ts = StringToTimestamp("1970-01-01 00:01:00");
   seconds = 60;
   EXPECT_EQ(ts, to_timestamp_int32(seconds));
   EXPECT_EQ(ts, to_timestamp_int64(seconds));
+  EXPECT_EQ(ts, to_timestamp_float32(seconds));
+  EXPECT_EQ(ts, to_timestamp_float64(seconds));
 
   ts = StringToTimestamp("1970-01-01 01:00:00");
   seconds = 60 * 60;
   EXPECT_EQ(ts, to_timestamp_int32(seconds));
   EXPECT_EQ(ts, to_timestamp_int64(seconds));
+  EXPECT_EQ(ts, to_timestamp_float32(seconds));
+  EXPECT_EQ(ts, to_timestamp_float64(seconds));
 
   ts = StringToTimestamp("1970-01-02 00:00:00");
   seconds = 24 * 60 * 60;
   EXPECT_EQ(ts, to_timestamp_int32(seconds));
   EXPECT_EQ(ts, to_timestamp_int64(seconds));
+  EXPECT_EQ(ts, to_timestamp_float32(seconds));
+  EXPECT_EQ(ts, to_timestamp_float64(seconds));
 }
 
 TEST(TestTime, TestToTimeNumeric) {
   auto millis = 0;
   EXPECT_EQ(millis, to_time_int32(millis));
   EXPECT_EQ(millis, to_time_int64(millis));
+  EXPECT_EQ(millis, to_time_float32(millis));
+  EXPECT_EQ(millis, to_time_float64(millis));
 
   millis = 1000;
   EXPECT_EQ(millis, to_time_int32(millis));
   EXPECT_EQ(millis, to_time_int64(millis));
+  EXPECT_EQ(millis, to_time_float32(millis));
+  EXPECT_EQ(millis, to_time_float64(millis));
 
   millis = 600000;
   EXPECT_EQ(millis, to_time_int32(millis));
   EXPECT_EQ(millis, to_time_int64(millis));
+  EXPECT_EQ(millis, to_time_float32(millis));
+  EXPECT_EQ(millis, to_time_float64(millis));
 
   millis = 360000;
   EXPECT_EQ(millis, to_time_int32(millis));
   EXPECT_EQ(millis, to_time_int64(millis));
+  EXPECT_EQ(millis, to_time_float32(millis));
+  EXPECT_EQ(millis, to_time_float64(millis));
 
   millis = 86400000;
   EXPECT_EQ(millis, to_time_int32(millis));
   EXPECT_EQ(millis, to_time_int64(millis));
+  EXPECT_EQ(millis, to_time_float32(millis));
+  EXPECT_EQ(millis, to_time_float64(millis));
 }
 
 }  // namespace gandiva
diff --git a/cpp/src/gandiva/precompiled/types.h b/cpp/src/gandiva/precompiled/types.h
@@ -101,9 +101,13 @@ gdv_timestamp add_date64_int64(gdv_date64, gdv_int64);
 
 gdv_timestamp to_timestamp_int32(gdv_int32);
 gdv_timestamp to_timestamp_int64(gdv_int64);
+gdv_timestamp to_timestamp_float32(gdv_float32);
+gdv_timestamp to_timestamp_float64(gdv_float64);
 
 gdv_time32 to_time_int32(gdv_int32);
 gdv_time32 to_time_int64(gdv_int64);
+gdv_time32 to_time_float32(gdv_float32);
+gdv_time32 to_time_float64(gdv_float64);
 
 gdv_int64 date_sub_timestamp_int32(gdv_timestamp, gdv_int32);
 gdv_int64 subtract_timestamp_int32(gdv_timestamp, gdv_int32);
