test cases for time

apache · Aug 15, 2022 · 839b777 · 839b777
1 parent 15a9a4b
commit 839b777
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/datafusion/core/tests/sql/timestamp.rs b/datafusion/core/tests/sql/timestamp.rs
@@ -1024,6 +1024,57 @@ async fn cast_string_to_time() {
     );
 }
 
+#[tokio::test]
+async fn time_comparison() -> Result<()> {
+    let ctx = SessionContext::new();
+
+    let sql = "SELECT TIME '00:00:00' = TIME '00:00:00'";
+    let results = execute_to_batches(&ctx, sql).await;
+    let expected = vec![
+        "+--------------------------------------------------------------------------------------------+",
+        "| CAST(Utf8(\"00:00:00\") AS Time64(Nanosecond)) = CAST(Utf8(\"00:00:00\") AS Time64(Nanosecond))|",
+        "+--------------------------------------------------------------------------------------------+",
+        "| true                                                                                       |",
+        "+--------------------------------------------------------------------------------------------+",
+    ];
+    assert_batches_eq!(expected, &results);
+
+    let sql = "SELECT TIME '00:00:00' = TIME '00:00:01'";
+    let results = execute_to_batches(&ctx, sql).await;
+    let expected = vec![
+        "+--------------------------------------------------------------------------------------------+",
+        "| CAST(Utf8(\"00:00:00\") AS Time64(Nanosecond)) = CAST(Utf8(\"00:00:01\") AS Time64(Nanosecond))|",
+        "+--------------------------------------------------------------------------------------------+",
+        "| false                                                                                      |",
+        "+--------------------------------------------------------------------------------------------+",
+    ];
+    assert_batches_eq!(expected, &results);
+
+    let sql = "SELECT TIME '00:00:00' < TIME '00:00:01'";
+    let results = execute_to_batches(&ctx, sql).await;
+    let expected = vec![
+        "+--------------------------------------------------------------------------------------------+",
+        "| CAST(Utf8(\"00:00:00\") AS Time64(Nanosecond)) < CAST(Utf8(\"00:00:01\") AS Time64(Nanosecond))|",
+        "+--------------------------------------------------------------------------------------------+",
+        "| true                                                                                       |",
+        "+--------------------------------------------------------------------------------------------+",
+    ];
+    assert_batches_eq!(expected, &results);
+
+    let sql = "SELECT TIME '00:00:00' > TIME '00:00:01'";
+    let results = execute_to_batches(&ctx, sql).await;
+    let expected = vec![
+        "+--------------------------------------------------------------------------------------------+",
+        "| CAST(Utf8(\"00:00:00\") AS Time64(Nanosecond)) > CAST(Utf8(\"00:00:01\") AS Time64(Nanosecond))|",
+        "+--------------------------------------------------------------------------------------------+",
+        "| false                                                                                      |",
+        "+--------------------------------------------------------------------------------------------+",
+    ];
+    assert_batches_eq!(expected, &results);
+
+    Ok(())
+}
+
 #[tokio::test]
 async fn cast_to_timestamp_twice() -> Result<()> {
     let ctx = SessionContext::new();