From 26f389594aa4db02cb6cc5223e708301c92d4e07 Mon Sep 17 00:00:00 2001 From: Darren Ranalli Date: Thu, 4 Sep 2003 00:53:08 +0000 Subject: [PATCH] generalize choose's rng --- direct/src/directutil/WeightedChoice.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/direct/src/directutil/WeightedChoice.py b/direct/src/directutil/WeightedChoice.py index 134f609a59..3038a32947 100755 --- 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]