(d)
| 2869 | @pytest.mark.acero |
| 2870 | def test_table_group_by(): |
| 2871 | def sorted_by_keys(d): |
| 2872 | # Ensure a guaranteed order of keys for aggregation results. |
| 2873 | if "keys2" in d: |
| 2874 | keys = tuple(zip(d["keys"], d["keys2"])) |
| 2875 | else: |
| 2876 | keys = d["keys"] |
| 2877 | sorted_keys = sorted(keys) |
| 2878 | sorted_d = {"keys": sorted(d["keys"])} |
| 2879 | for entry in d: |
| 2880 | if entry == "keys": |
| 2881 | continue |
| 2882 | values = dict(zip(keys, d[entry])) |
| 2883 | for k in sorted_keys: |
| 2884 | sorted_d.setdefault(entry, []).append(values[k]) |
| 2885 | return sorted_d |
| 2886 | |
| 2887 | table = pa.table([ |
| 2888 | pa.array(["a", "a", "b", "b", "c"]), |
no test coverage detected