You are viewing a plain text version of this content. The canonical link for it is here.
Posted to github@arrow.apache.org by GitBox <gi...@apache.org> on 2022/04/13 02:59:50 UTC
[GitHub] [arrow-datafusion] Ted-Jiang commented on a diff in pull request #2192: [Ballista] Enable ApproxPercentileWithWeight in Ballista and fill UT
Ted-Jiang commented on code in PR #2192:
URL: https://github.com/apache/arrow-datafusion/pull/2192#discussion_r849028186
##########
ballista/rust/client/src/context.rs:
##########
@@ -718,6 +719,199 @@ mod tests {
)
.await
.unwrap();
+
+ let df = context.sql("select min(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------+",
+ "| MIN(test.id) |",
+ "+--------------+",
+ "| 0 |",
+ "+--------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context.sql("select max(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------+",
+ "| MAX(test.id) |",
+ "+--------------+",
+ "| 7 |",
+ "+--------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context.sql("select SUM(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------+",
+ "| SUM(test.id) |",
+ "+--------------+",
+ "| 28 |",
+ "+--------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context.sql("select AVG(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------+",
+ "| AVG(test.id) |",
+ "+--------------+",
+ "| 3.5 |",
+ "+--------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context.sql("select COUNT(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+----------------+",
+ "| COUNT(test.id) |",
+ "+----------------+",
+ "| 8 |",
+ "+----------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select approx_distinct(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+-------------------------+",
+ "| APPROXDISTINCT(test.id) |",
+ "+-------------------------+",
+ "| 8 |",
+ "+-------------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select ARRAY_AGG(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------------------+",
+ "| ARRAYAGG(test.id) |",
+ "+--------------------------+",
+ "| [4, 5, 6, 7, 2, 3, 0, 1] |",
+ "+--------------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context.sql("select VAR(\"id\") from test").await.unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+-------------------+",
+ "| VARIANCE(test.id) |",
+ "+-------------------+",
+ "| 6.000000000000001 |",
+ "+-------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select VAR_POP(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+----------------------+",
+ "| VARIANCEPOP(test.id) |",
+ "+----------------------+",
+ "| 5.250000000000001 |",
+ "+----------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select VAR_SAMP(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+-------------------+",
+ "| VARIANCE(test.id) |",
+ "+-------------------+",
+ "| 6.000000000000001 |",
+ "+-------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select STDDEV(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------------+",
+ "| STDDEV(test.id) |",
+ "+--------------------+",
+ "| 2.4494897427831783 |",
+ "+--------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select STDDEV_SAMP(\"id\") from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------------+",
+ "| STDDEV(test.id) |",
+ "+--------------------+",
+ "| 2.4494897427831783 |",
+ "+--------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select COVAR(id, tinyint_col) from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+--------------------------------------+",
+ "| COVARIANCE(test.id,test.tinyint_col) |",
+ "+--------------------------------------+",
+ "| 0.28571428571428586 |",
+ "+--------------------------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select CORR(id, tinyint_col) from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+---------------------------------------+",
+ "| CORRELATION(test.id,test.tinyint_col) |",
+ "+---------------------------------------+",
+ "| 0.21821789023599245 |",
+ "+---------------------------------------+",
+ ];
+ assert_result_eq(expected, &*res);
+
+ let df = context
+ .sql("select approx_percentile_cont_with_weight(\"id\", 2, 0.5) from test")
+ .await
+ .unwrap();
+ let res = df.collect().await.unwrap();
+ let expected = vec![
+ "+---------------------------------------------------------------+",
+ "| APPROXPERCENTILECONTWITHWEIGHT(test.id,Int64(2),Float64(0.5)) |",
Review Comment:
I thinks this name is align with proto file, it try to rewrite `debug` or `display` of this struct , it not works😂
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org