generalize choose's rng

2003-09-04 00:53:08 +00:00 · 2003-09-04 00:53:08 +00:00 · 26f389594a
parent 71839729a8
commit 26f389594a
1 changed files with 2 additions and 2 deletions
--- a/direct/src/directutil/WeightedChoice.py
+++ b/direct/src/directutil/WeightedChoice.py
@ -10,8 +10,8 @@ class WeightedChoice:
        self.listOfLists = listOfLists
        self.weightIndex = weightIndex
   
-    def choose(self):
-        roll = random.randrange(self.total)
+    def choose(self, rng=random):
+        roll = rng.randrange(self.total)
        weight = self.weightIndex
        for i in self.listOfLists:
            roll -= i[weight]