From 620d78f43b2dbdfb0fb6711b8257edcedbb557e7 Mon Sep 17 00:00:00 2001
From: Thomas Dillow <tdillow@ucsc.edu>
Date: Thu, 13 Mar 2025 22:08:00 -0700
Subject: [PATCH] basic shuffle revisions
---
tests/shuffle/basic_shuffle.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py
index a8fbd5f..c23cda6 100644
--- a/tests/shuffle/basic_shuffle.py
+++ b/tests/shuffle/basic_shuffle.py
@@ -86,15 +86,18 @@ def basic_shuffle(conductor: ClusterConductor, dir, log: Logger):
assert len(shard1_keys) + len(shard2_keys) == 15, f"expected 15 keys, got {len(shard1_keys) + len(shard2_keys)}"
+ # Remove shard 2. This loses keys.
+ conductor.remove_shard("shard2")
fx.broadcast_view(conductor.get_shard_view())
r = c.get_all(0, timeout=10)
assert r.ok, f"expected ok for get, got {r.status_code}"
res = r.json()["items"]
shard1_keys_after_delete = res
-
+
assert len(shard1_keys) == len(shard1_keys_after_delete), f"expected {len(shard1_keys)} keys, got {len(shard1_keys_after_delete)}"
+
return True, "ok"
def partitioned_shards(conductor: ClusterConductor, dir, log: Logger):
--
GitLab