From 11a3185603a821285a055699ab55519424f7f38f Mon Sep 17 00:00:00 2001 From: Marco Federighi Date: Fri, 19 Dec 2014 12:58:54 +0100 Subject: [PATCH] Add PositiveDecimalGenerator --- autofixture/generators.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/autofixture/generators.py b/autofixture/generators.py index cf57738..8fad8ed 100644 --- a/autofixture/generators.py +++ b/autofixture/generators.py @@ -314,6 +314,18 @@ def generate(self): return value +class PositiveDecimalGenerator(DecimalGenerator): + def __init__(self, max_digits=None, decimal_places=None, *args, **kwargs): + super(PositiveDecimalGenerator, self).__init__(max_digits, decimal_places, *args, **kwargs) + + def generate(self): + maxint = 10 ** self.max_digits - 1 + value = ( + Decimal(random.randint(0, maxint)) / + 10 ** self.decimal_places) + return value + + class FirstNameGenerator(Generator): """ Generates a first name, either male or female """