update UT's

datafusion-contrib · Sep 27, 2024 · 3c47a45 · 3c47a45
1 parent ad1910c
commit 3c47a45
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/tests/main.rs b/tests/main.rs
@@ -252,9 +252,10 @@ async fn test_max_by_and_min_by() {
 }
 
 #[tokio::test]
-async fn test_kurtosis_pop_int64() {
+async fn test_kurtosis_pop() {
     let mut execution = TestExecution::new().await.unwrap().with_setup(TEST_TABLE).await;
 
+    // Test with int64
     let actual = execution
         .run_and_format("SELECT kurtosis_pop(int64_col) FROM test_table")
         .await;
@@ -266,21 +267,17 @@ async fn test_kurtosis_pop_int64() {
         - "| -0.9599999999999755                |"
         - +------------------------------------+
     "###);
-}
-
-#[tokio::test]
-async fn test_kurtosis_pop_float64() {
-    let mut execution = TestExecution::new().await.unwrap().with_setup(TEST_TABLE).await;
 
+    // Test with float64
     let actual = execution
         .run_and_format("SELECT kurtosis_pop(float64_col) FROM test_table")
         .await;
 
     insta::assert_yaml_snapshot!(actual, @r###"
-        - +--------------------------------------+
-        - "| kurtosis_pop(test_table.float64_col) |"
-        - +--------------------------------------+
-        - "| -0.9599999999999755                  |"
-        - +--------------------------------------+
-    "###);
+    - +--------------------------------------+
+    - "| kurtosis_pop(test_table.float64_col) |"
+    - +--------------------------------------+
+    - "| -0.9599999999999755                  |"
+    - +--------------------------------------+
+"###);
 }