(d)
| 2822 | @pytest.mark.acero |
| 2823 | def test_table_group_by(): |
| 2824 | def sorted_by_keys(d): |
| 2825 | # Ensure a guaranteed order of keys for aggregation results. |
| 2826 | if "keys2" in d: |
| 2827 | keys = tuple(zip(d["keys"], d["keys2"])) |
| 2828 | else: |
| 2829 | keys = d["keys"] |
| 2830 | sorted_keys = sorted(keys) |
| 2831 | sorted_d = {"keys": sorted(d["keys"])} |
| 2832 | for entry in d: |
| 2833 | if entry == "keys": |
| 2834 | continue |
| 2835 | values = dict(zip(keys, d[entry])) |
| 2836 | for k in sorted_keys: |
| 2837 | sorted_d.setdefault(entry, []).append(values[k]) |
| 2838 | return sorted_d |
| 2839 | |
| 2840 | table = pa.table([ |
| 2841 | pa.array(["a", "a", "b", "b", "c"]), |
no test coverage detected