add tests

XiangpengHao · Jul 26, 2024 · 6bdbb72 · 6bdbb72
1 parent ba2a2f1
commit 6bdbb72
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/datafusion/functions/src/utils.rs b/datafusion/functions/src/utils.rs
@@ -183,6 +183,21 @@ pub mod test {
         };
     }
 
+    use arrow::datatypes::DataType;
     #[allow(unused_imports)]
     pub(crate) use test_function;
+
+    use super::*;
+
+    #[test]
+    fn string_to_int_type() {
+        let v = utf8_to_int_type(&DataType::Utf8, "test").unwrap();
+        assert_eq!(v, DataType::Int32);
+
+        let v = utf8_to_int_type(&DataType::Utf8View, "test").unwrap();
+        assert_eq!(v, DataType::Int32);
+
+        let v = utf8_to_int_type(&DataType::LargeUtf8, "test").unwrap();
+        assert_eq!(v, DataType::Int64);
+    }
 }