From 6fc46fdac87660b0736f65c6ec8b88706e2a59a3 Mon Sep 17 00:00:00 2001
From: Alexander Aghili <alexander.w.aghili@gmail.com>
Date: Fri, 14 Mar 2025 11:01:55 -0700
Subject: [PATCH] fix basic shuffle 2

---
 tests/shuffle/basic_shuffle.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/shuffle/basic_shuffle.py b/tests/shuffle/basic_shuffle.py
index 898061c..6ba0e9d 100644
--- a/tests/shuffle/basic_shuffle.py
+++ b/tests/shuffle/basic_shuffle.py
@@ -215,7 +215,7 @@ def basic_shuffle_2(conductor: ClusterConductor, dir, log: Logger):
 
         node_to_put = 0
         base_key = "key"
-        for i in range(0, 50):
+        for i in range(0, 300):
             r = c.put(node_to_put, f"{base_key}{i}", f"{i}", timeout=10)
             assert r.ok, f"expected ok for new key, got {r.status_code}"
             node_to_put += 1
@@ -243,7 +243,7 @@ def basic_shuffle_2(conductor: ClusterConductor, dir, log: Logger):
         for key in keys2:
             assert not (key in keys1)
         
-        assert original_get_all.keys() == keys1 + keys2
+        assert original_get_all.keys() == keys1 | keys2
         assert len(original_get_all) == len(keys1) + len(keys2)
 
         return True, "ok"
-- 
GitLab