|
17 | 17 | #include <cudf/table/table.hpp> |
18 | 18 | #include <cudf/types.hpp> |
19 | 19 |
|
20 | | -namespace cudf ::test { |
21 | | - |
22 | | -void test_single_agg(column_view const& keys, |
23 | | - column_view const& values, |
24 | | - column_view const& expect_keys, |
25 | | - column_view const& expect_vals, |
26 | | - std::unique_ptr<groupby_aggregation>&& agg, |
| 20 | +void test_single_agg(cudf::column_view const& keys, |
| 21 | + cudf::column_view const& values, |
| 22 | + cudf::column_view const& expect_keys, |
| 23 | + cudf::column_view const& expect_vals, |
| 24 | + std::unique_ptr<cudf::groupby_aggregation>&& agg, |
27 | 25 | force_use_sort_impl use_sort, |
28 | | - null_policy include_null_keys, |
29 | | - sorted keys_are_sorted, |
30 | | - std::vector<order> const& column_order, |
31 | | - std::vector<null_order> const& null_precedence, |
32 | | - sorted reference_keys_are_sorted, |
| 26 | + cudf::null_policy include_null_keys, |
| 27 | + cudf::sorted keys_are_sorted, |
| 28 | + std::vector<cudf::order> const& column_order, |
| 29 | + std::vector<cudf::null_order> const& null_precedence, |
| 30 | + cudf::sorted reference_keys_are_sorted, |
33 | 31 | std::source_location const& location) |
34 | 32 | { |
35 | 33 | SCOPED_TRACE("Original failure location: " + std::string{location.file_name()} + ":" + |
36 | 34 | std::to_string(location.line())); |
37 | 35 |
|
38 | 36 | auto const [sorted_expect_keys, sorted_expect_vals] = [&]() { |
39 | | - if (reference_keys_are_sorted == sorted::NO) { |
| 37 | + if (reference_keys_are_sorted == cudf::sorted::NO) { |
40 | 38 | auto const sort_expect_order = |
41 | | - sorted_order(table_view{{expect_keys}}, column_order, null_precedence); |
42 | | - auto sorted_expect_keys = gather(table_view{{expect_keys}}, *sort_expect_order); |
43 | | - auto sorted_expect_vals = gather(table_view{{expect_vals}}, *sort_expect_order); |
| 39 | + sorted_order(cudf::table_view{{expect_keys}}, column_order, null_precedence); |
| 40 | + auto sorted_expect_keys = gather(cudf::table_view{{expect_keys}}, *sort_expect_order); |
| 41 | + auto sorted_expect_vals = gather(cudf::table_view{{expect_vals}}, *sort_expect_order); |
44 | 42 | return std::make_pair(std::move(sorted_expect_keys), std::move(sorted_expect_vals)); |
45 | 43 | } |
46 | | - auto sorted_expect_keys = std::make_unique<table>(table_view{{expect_keys}}); |
47 | | - auto sorted_expect_vals = std::make_unique<table>(table_view{{expect_vals}}); |
| 44 | + auto sorted_expect_keys = std::make_unique<cudf::table>(cudf::table_view{{expect_keys}}); |
| 45 | + auto sorted_expect_vals = std::make_unique<cudf::table>(cudf::table_view{{expect_vals}}); |
48 | 46 | return std::make_pair(std::move(sorted_expect_keys), std::move(sorted_expect_vals)); |
49 | 47 | }(); |
50 | 48 |
|
51 | | - std::vector<groupby::aggregation_request> requests; |
| 49 | + std::vector<cudf::groupby::aggregation_request> requests; |
52 | 50 | requests.emplace_back(); |
53 | 51 | requests[0].values = values; |
54 | 52 |
|
55 | 53 | requests[0].aggregations.push_back(std::move(agg)); |
56 | 54 |
|
57 | 55 | if (use_sort == force_use_sort_impl::YES) { |
58 | 56 | // WAR to force groupby to use sort implementation |
59 | | - requests[0].aggregations.push_back(make_nth_element_aggregation<groupby_aggregation>(0)); |
| 57 | + requests[0].aggregations.push_back( |
| 58 | + cudf::make_nth_element_aggregation<cudf::groupby_aggregation>(0)); |
60 | 59 | } |
61 | 60 |
|
62 | 61 | // since the default behavior of groupby(...) for an empty null_precedence vector is |
63 | 62 | // null_order::AFTER whereas for sorted_order(...) it's null_order::BEFORE |
64 | | - auto const precedence = |
65 | | - null_precedence.empty() ? std::vector<null_order>(1, null_order::BEFORE) : null_precedence; |
| 63 | + auto const precedence = null_precedence.empty() |
| 64 | + ? std::vector<cudf::null_order>(1, cudf::null_order::BEFORE) |
| 65 | + : null_precedence; |
66 | 66 |
|
67 | | - groupby::groupby gb_obj( |
68 | | - table_view({keys}), include_null_keys, keys_are_sorted, column_order, precedence); |
| 67 | + cudf::groupby::groupby gb_obj( |
| 68 | + cudf::table_view({keys}), include_null_keys, keys_are_sorted, column_order, precedence); |
69 | 69 |
|
70 | | - auto result = gb_obj.aggregate(requests, test::get_default_stream()); |
| 70 | + auto result = gb_obj.aggregate(requests, cudf::test::get_default_stream()); |
71 | 71 |
|
72 | | - if (use_sort == force_use_sort_impl::YES && keys_are_sorted == sorted::NO) { |
| 72 | + if (use_sort == force_use_sort_impl::YES && keys_are_sorted == cudf::sorted::NO) { |
73 | 73 | CUDF_TEST_EXPECT_TABLES_EQUAL(*sorted_expect_keys, result.first->view()); |
74 | 74 | CUDF_TEST_EXPECT_COLUMNS_EQUIVALENT(sorted_expect_vals->get_column(0), |
75 | 75 | *result.second[0].results[0]); |
76 | 76 |
|
77 | 77 | } else { |
78 | 78 | auto const sort_order = sorted_order(result.first->view(), column_order, precedence); |
79 | 79 | auto const sorted_keys = gather(result.first->view(), *sort_order); |
80 | | - auto const sorted_vals = gather(table_view({result.second[0].results[0]->view()}), *sort_order); |
| 80 | + auto const sorted_vals = |
| 81 | + gather(cudf::table_view({result.second[0].results[0]->view()}), *sort_order); |
81 | 82 |
|
82 | 83 | CUDF_TEST_EXPECT_TABLES_EQUAL(*sorted_expect_keys, *sorted_keys); |
83 | 84 | CUDF_TEST_EXPECT_COLUMNS_EQUIVALENT(sorted_expect_vals->get_column(0), |
84 | 85 | sorted_vals->get_column(0)); |
85 | 86 | } |
86 | 87 | } |
87 | 88 |
|
88 | | -void test_sum_agg(column_view const& keys, |
89 | | - column_view const& values, |
90 | | - column_view const& expected_keys, |
91 | | - column_view const& expected_values, |
| 89 | +void test_sum_agg(cudf::column_view const& keys, |
| 90 | + cudf::column_view const& values, |
| 91 | + cudf::column_view const& expected_keys, |
| 92 | + cudf::column_view const& expected_values, |
92 | 93 | std::source_location const& location) |
93 | 94 | { |
94 | 95 | auto const do_test = [&](auto const use_sort_option) { |
95 | 96 | test_single_agg(keys, |
96 | 97 | values, |
97 | 98 | expected_keys, |
98 | 99 | expected_values, |
99 | | - make_sum_aggregation<groupby_aggregation>(), |
| 100 | + cudf::make_sum_aggregation<cudf::groupby_aggregation>(), |
100 | 101 | use_sort_option, |
101 | | - null_policy::INCLUDE, |
102 | | - sorted::NO, |
| 102 | + cudf::null_policy::INCLUDE, |
| 103 | + cudf::sorted::NO, |
103 | 104 | {}, |
104 | 105 | {}, |
105 | | - sorted::NO, |
| 106 | + cudf::sorted::NO, |
106 | 107 | location); |
107 | 108 | }; |
108 | 109 | do_test(force_use_sort_impl::YES); |
109 | 110 | do_test(force_use_sort_impl::NO); |
110 | 111 | } |
111 | 112 |
|
112 | | -void test_single_scan(column_view const& keys, |
113 | | - column_view const& values, |
114 | | - column_view const& expect_keys, |
115 | | - column_view const& expect_vals, |
116 | | - std::unique_ptr<groupby_scan_aggregation>&& agg, |
117 | | - null_policy include_null_keys, |
118 | | - sorted keys_are_sorted, |
119 | | - std::vector<order> const& column_order, |
120 | | - std::vector<null_order> const& null_precedence, |
| 113 | +void test_single_scan(cudf::column_view const& keys, |
| 114 | + cudf::column_view const& values, |
| 115 | + cudf::column_view const& expect_keys, |
| 116 | + cudf::column_view const& expect_vals, |
| 117 | + std::unique_ptr<cudf::groupby_scan_aggregation>&& agg, |
| 118 | + cudf::null_policy include_null_keys, |
| 119 | + cudf::sorted keys_are_sorted, |
| 120 | + std::vector<cudf::order> const& column_order, |
| 121 | + std::vector<cudf::null_order> const& null_precedence, |
121 | 122 | std::source_location const& location) |
122 | 123 | { |
123 | 124 | SCOPED_TRACE("Original failure location: " + std::string{location.file_name()} + ":" + |
124 | 125 | std::to_string(location.line())); |
125 | 126 |
|
126 | | - std::vector<groupby::scan_request> requests; |
| 127 | + std::vector<cudf::groupby::scan_request> requests; |
127 | 128 | requests.emplace_back(); |
128 | 129 | requests[0].values = values; |
129 | 130 | requests[0].aggregations.push_back(std::move(agg)); |
130 | 131 |
|
131 | | - groupby::groupby gb_obj( |
132 | | - table_view({keys}), include_null_keys, keys_are_sorted, column_order, null_precedence); |
| 132 | + cudf::groupby::groupby gb_obj( |
| 133 | + cudf::table_view({keys}), include_null_keys, keys_are_sorted, column_order, null_precedence); |
133 | 134 |
|
134 | 135 | // groupby scan uses sort implementation |
135 | 136 | auto result = gb_obj.scan(requests); |
136 | 137 |
|
137 | | - CUDF_TEST_EXPECT_TABLES_EQUAL(table_view({expect_keys}), result.first->view()); |
| 138 | + CUDF_TEST_EXPECT_TABLES_EQUAL(cudf::table_view({expect_keys}), result.first->view()); |
138 | 139 | CUDF_TEST_EXPECT_COLUMNS_EQUIVALENT(expect_vals, *result.second[0].results[0]); |
139 | 140 | } |
140 | | - |
141 | | -} // namespace cudf::test |
0 commit comments