pitrou commented on code in PR #45259:
URL: https://github.com/apache/arrow/pull/45259#discussion_r1923343948
##########
cpp/src/arrow/compute/kernels/vector_sort_test.cc:
##########
@@ -2466,5 +2469,151 @@ TEST_F(TestRank, EmptyChunks) {
}
}
+class TestRankQuantile : public BaseTestRank {
+ public:
+ void AssertRankQuantile(const DatumVector& datums, SortOrder order,
+ NullPlacement null_placement, double factor,
+ const std::shared_ptr<Array>& expected) {
+ const std::vector<SortKey> sort_keys{SortKey("foo", order)};
+ RankQuantileOptions options(sort_keys, null_placement, factor);
+ ARROW_SCOPED_TRACE("options = ", options.ToString());
+ for (const auto& datum : datums) {
+ ASSERT_OK_AND_ASSIGN(auto actual, CallFunction("rank_quantile", {datum},
&options));
+ ValidateOutput(actual);
+ AssertDatumsEqual(expected, actual, /*verbose=*/true);
+ }
+ }
+
+ void AssertRankQuantile(const DatumVector& datums, SortOrder order,
+ NullPlacement null_placement, double factor,
+ const std::string& expected) {
+ AssertRankQuantile(datums, order, null_placement, factor,
+ ArrayFromJSON(float64(), expected));
+ }
+
+ void AssertRankQuantile(SortOrder order, NullPlacement null_placement,
double factor,
+ const std::shared_ptr<Array>& expected) {
+ AssertRankQuantile(datums_, order, null_placement, factor, expected);
+ }
+
+ void AssertRankQuantile(SortOrder order, NullPlacement null_placement,
double factor,
+ const std::string& expected) {
+ AssertRankQuantile(datums_, order, null_placement, factor,
+ ArrayFromJSON(float64(), expected));
+ }
+
+ void AssertRankQuantileEmpty(std::shared_ptr<DataType> type) {
+ for (auto null_placement : AllNullPlacements()) {
+ for (auto order : AllOrders()) {
+ AssertRankQuantile({ArrayFromJSON(type, "[]")}, order, null_placement,
+ /*factor=*/1.0, "[]");
+ AssertRankQuantile({ArrayFromJSON(type, "[null]")}, order,
null_placement,
+ /*factor=*/1.0, "[0.5]");
+ AssertRankQuantile({ArrayFromJSON(type, "[null]")}, order,
null_placement,
+ /*factor=*/10.0, "[5]");
+ AssertRankQuantile({ArrayFromJSON(type, "[null, null, null]")}, order,
+ null_placement, /*factor=*/1.0, "[0.5, 0.5, 0.5]");
+ AssertRankQuantile({ArrayFromJSON(type, "[null, null, null]")}, order,
+ null_placement, /*factor=*/100.0, "[50, 50, 50]");
+ }
+ }
+ }
+
+ // Expecting an input ordered like [1, 2, 1, 2, 1]
+ void AssertRankQuantile_12121() {
+ for (auto null_placement : AllNullPlacements()) {
+ AssertRankQuantile(SortOrder::Ascending, null_placement, 100.0,
+ "[30.0, 80.0, 30.0, 80.0, 30.0]");
+ AssertRankQuantile(SortOrder::Descending, null_placement, 100.0,
Review Comment:
Done!
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
