Skip to content

Commit

Permalink
Remove outdated test
Browse files Browse the repository at this point in the history
  • Loading branch information
hendrikmakait committed Sep 13, 2024
1 parent 3cdee06 commit 1c0a178
Showing 1 changed file with 0 additions and 27 deletions.
27 changes: 0 additions & 27 deletions distributed/shuffle/tests/test_shuffle.py
Original file line number Diff line number Diff line change
Expand Up @@ -2685,33 +2685,6 @@ async def barrier(self, id: ShuffleId, run_id: int, consistent: bool) -> None:
return await super().barrier(id, run_id, consistent)


@gen_cluster(client=True)
async def test_unpack_is_non_rootish(c, s, a, b):
with pytest.warns(UserWarning):
scheduler_plugin = BlockedBarrierShuffleSchedulerPlugin(s)
df = dask.datasets.timeseries(
start="2000-01-01",
end="2000-01-21",
dtypes={"x": float, "y": float},
freq="10 s",
)
df = df.shuffle("x")
result = c.compute(df)

await scheduler_plugin.in_barrier.wait()

unpack_tss = [ts for key, ts in s.tasks.items() if key_split(key) == UNPACK_PREFIX]
assert len(unpack_tss) == 20
assert not any(s.is_rootish(ts) for ts in unpack_tss)
del unpack_tss
scheduler_plugin.block_barrier.set()
result = await result

await assert_worker_cleanup(a)
await assert_worker_cleanup(b)
await assert_scheduler_cleanup(s)


class FlakyConnectionPool(ConnectionPool):
def __init__(self, *args, failing_connects=0, **kwargs):
self.attempts = 0
Expand Down

0 comments on commit 1c0a178

Please sign in to comment.