From a53af6c87540c41fdf34211ea99ee794cae6d160 Mon Sep 17 00:00:00 2001
From: Damien George
Date: Fri, 22 Jan 2016 16:19:32 +0000
Subject: extmod/modurandom: Add some extra random functions.

Functions added are:
- randint
- randrange
- choice
- random
- uniform

They are enabled with configuration variable
MICROPY_PY_URANDOM_EXTRA_FUNCS, which is disabled by default.  It is
enabled for unix coverage build and stmhal.
---
 py/mpconfig.h | 5 +++++
 py/qstrdefs.h | 7 +++++++
 2 files changed, 12 insertions(+)

(limited to 'py')

diff --git a/py/mpconfig.h b/py/mpconfig.h
index 72af343dd..402e97acc 100644
--- a/py/mpconfig.h
+++ b/py/mpconfig.h
@@ -787,6 +787,11 @@ typedef double mp_float_t;
 #define MICROPY_PY_URANDOM (0)
 #endif
 
+// Whether to include: randrange, randint, choice, random, uniform
+#ifndef MICROPY_PY_URANDOM_EXTRA_FUNCS
+#define MICROPY_PY_URANDOM_EXTRA_FUNCS (0)
+#endif
+
 #ifndef MICROPY_PY_MACHINE
 #define MICROPY_PY_MACHINE (0)
 #endif
diff --git a/py/qstrdefs.h b/py/qstrdefs.h
index dd76c3d43..122fca128 100644
--- a/py/qstrdefs.h
+++ b/py/qstrdefs.h
@@ -686,4 +686,11 @@ Q(dupterm)
 Q(urandom)
 Q(getrandbits)
 Q(seed)
+#if MICROPY_PY_URANDOM_EXTRA_FUNCS
+Q(randrange)
+Q(randint)
+Q(choice)
+Q(random)
+Q(uniform)
+#endif
 #endif
-- 
cgit v1.2.3