(c, s, *workers)
| 941 | |
| 942 | @gen_cluster(client=True) |
| 943 | async def _test(c, s, *workers): |
| 944 | test_df = dd.from_dict( |
| 945 | { |
| 946 | "partition": [0, 1, 2, 3, 0, 1, 2, 3], |
| 947 | "value": [1, 2, 3, 4, 5, 6, 7, 8], |
| 948 | }, |
| 949 | npartitions=2, |
| 950 | ) |
| 951 | part_df = test_df.shuffle("partition", force=True, shuffle_method="p2p") |
| 952 | |
| 953 | @delayed |
| 954 | def delayed_func(x): |
| 955 | return x |
| 956 | |
| 957 | delay_df = delayed_func(part_df) |
| 958 | dd.utils.assert_eq( |
| 959 | await c.gather(c.compute(delay_df)), |
| 960 | await c.gather(c.compute(part_df)), |
| 961 | ) |
| 962 | |
| 963 | _test() |
no test coverage detected