| 2262 | }; |
| 2263 | |
| 2264 | class TestRank : public BaseTestRank { |
| 2265 | protected: |
| 2266 | static void AssertRank(const DatumVector& datums, SortOrder order, |
| 2267 | NullPlacement null_placement, RankOptions::Tiebreaker tiebreaker, |
| 2268 | const std::shared_ptr<Array>& expected) { |
| 2269 | const std::vector<SortKey> sort_keys{SortKey("foo", order)}; |
| 2270 | RankOptions options(sort_keys, null_placement, tiebreaker); |
| 2271 | ARROW_SCOPED_TRACE("options = ", options.ToString()); |
| 2272 | for (const auto& datum : datums) { |
| 2273 | ASSERT_OK_AND_ASSIGN(auto actual, CallFunction("rank", {datum}, &options)); |
| 2274 | ValidateOutput(actual); |
| 2275 | AssertDatumsEqual(expected, actual, /*verbose=*/true); |
| 2276 | } |
| 2277 | } |
| 2278 | |
| 2279 | void AssertRank(SortOrder order, NullPlacement null_placement, |
| 2280 | RankOptions::Tiebreaker tiebreaker, |
| 2281 | const std::shared_ptr<Array>& expected) const { |
| 2282 | AssertRank(datums_, order, null_placement, tiebreaker, expected); |
| 2283 | } |
| 2284 | |
| 2285 | static void AssertRankEmpty(std::shared_ptr<DataType> type, SortOrder order, |
| 2286 | NullPlacement null_placement, |
no outgoing calls
no test coverage detected