more UT's

datafusion-contrib · Sep 27, 2024 · be8b69b · be8b69b
1 parent 3c47a45
commit be8b69b
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -77,13 +77,6 @@ SELECT min_by(x, y) FROM VALUES (1, 10), (2, 5), (3, 15), (4, 8) as tab(x, y);
 -- | 2                   |
 -- +---------------------+
 
-SELECT kurtosis_pop(col) FROM VALUES (1.0), (10.0), (100.0), (10.0), (1.0) as tab(col);
--- Results in
--- +-----------------------+
--- | kurtosis_pop(tab.col) |
--- +-----------------------+
--- | 0.194323231917        |
--- +-----------------------+
 ```
 
 ## Done

diff --git a/src/lib.rs b/src/lib.rs
@@ -30,6 +30,7 @@ pub mod kurtosis_pop;
 pub mod max_min_by;
 pub mod mode;
 pub mod expr_extra_fn {
+    pub use super::kurtosis_pop::kurtosis_pop;
     pub use super::max_min_by::max_by;
     pub use super::max_min_by::min_by;
     pub use super::mode::mode;

diff --git a/tests/main.rs b/tests/main.rs
@@ -279,5 +279,34 @@ async fn test_kurtosis_pop() {
     - +--------------------------------------+
     - "| -0.9599999999999755                  |"
     - +--------------------------------------+
+"###);
+
+    let actual = execution
+        .run_and_format("SELECT kurtosis_pop(col) FROM VALUES (1.0) as tab(col)")
+        .await;
+    insta::assert_yaml_snapshot!(actual, @r###"
+    - +-----------------------+
+    - "| kurtosis_pop(tab.col) |"
+    - +-----------------------+
+    - "|                       |"
+    - +-----------------------+
+"###);
+
+    let actual = execution.run_and_format("SELECT kurtosis_pop(1.0)").await;
+    insta::assert_yaml_snapshot!(actual, @r###"
+    - +--------------------------+
+    - "| kurtosis_pop(Float64(1)) |"
+    - +--------------------------+
+    - "|                          |"
+    - +--------------------------+
+"###);
+
+    let actual = execution.run_and_format("SELECT kurtosis_pop(null)").await;
+    insta::assert_yaml_snapshot!(actual, @r###"
+- +--------------------+
+- "| kurtosis_pop(NULL) |"
+- +--------------------+
+- "|                    |"
+- +--------------------+
 "###);
 }