From 02e4cfd9bb0e0d1124385f52108bb709fc5d72bf Mon Sep 17 00:00:00 2001 From: Ilya Pirogov Date: Fri, 27 Dec 2013 13:07:34 +0400 Subject: Added FuzzyInteger support for step --- factory/fuzzy.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'factory') diff --git a/factory/fuzzy.py b/factory/fuzzy.py index 34949c5..2ea544a 100644 --- a/factory/fuzzy.py +++ b/factory/fuzzy.py @@ -107,18 +107,19 @@ class FuzzyChoice(BaseFuzzyAttribute): class FuzzyInteger(BaseFuzzyAttribute): """Random integer within a given range.""" - def __init__(self, low, high=None, **kwargs): + def __init__(self, low, high=None, step=1, **kwargs): if high is None: high = low low = 0 self.low = low self.high = high + self.step = step super(FuzzyInteger, self).__init__(**kwargs) def fuzz(self): - return random.randint(self.low, self.high) + return random.randrange(self.low, self.high + 1, self.step) class FuzzyDecimal(BaseFuzzyAttribute): -- cgit v1.2.3