a RandomSet implementation useful for swapping into topological

1 files changed, 15 insertions, 1 deletions

diff --git a/lib/sqlalchemy/test/util.py b/lib/sqlalchemy/test/util.py
index 8a3a0e745..cd73b44d0 100644
--- a/lib/sqlalchemy/test/util.py
+++ b/lib/sqlalchemy/test/util.py
@@ -2,6 +2,7 @@ from sqlalchemy.util import jython, function_named
 
 import gc
 import time
+import random
 
 if jython:
     def gc_collect(*args):
@@ -50,4 +51,17 @@ def round_decimal(value, prec):
     # can also use shift() here but that is 2.6 only
     return (value * decimal.Decimal("1" + "0" * prec)).to_integral(decimal.ROUND_FLOOR) / \
                         pow(10, prec)
-    
\ No newline at end of file
+    
+class RandomSet(set):
+    def __iter__(self):
+        l = list(set.__iter__(self))
+        random.shuffle(l)
+        return iter(l)
+    
+    def pop(self):
+        index = random.randint(0, len(self) - 1)
+        item = list(set.__iter__(self))[index]
+        self.remove(item)
+        return item
+        
+        
\ No newline at end of file