Merge pull request #14 from paulshannon/patch-1

Change IntegerField and PositiveIntegerField to smaller integer provider.
Brobin · Jun 29, 2015 · b17b7b2 · b17b7b2
2 parents 7072f94 + 0408c50
commit b17b7b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/django_seed/guessers.py b/django_seed/guessers.py
@@ -68,8 +68,8 @@ def guess_format(self, field):
         if isinstance(field, PositiveSmallIntegerField): return lambda x: provider.rand_small_int(pos=True)
         if isinstance(field, SmallIntegerField): return lambda x: provider.rand_small_int()
         if isinstance(field, BigIntegerField): return lambda x: provider.rand_big_int()
-        if isinstance(field, PositiveIntegerField): return lambda x: provider.rand_int(pos=True)
-        if isinstance(field, IntegerField): return lambda x: provider.rand_int()
+        if isinstance(field, PositiveIntegerField): return lambda x: provider.rand_small_int(pos=True)
+        if isinstance(field, IntegerField): return lambda x: provider.rand_small_int()
         if isinstance(field, FloatField): return lambda x: provider.rand_float()
         if isinstance(field, DecimalField): return lambda x: random.random()