tests: Replace umodule with module everywhere.

This work was funded through GitHub Sponsors.

Signed-off-by: Jim Mussared <jim.mussared@gmail.com>

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/extmod/random_basic.py b/tests/extmod/random_basic.py
new file mode 100644
index 000000000..7cb992fed
--- /dev/null
+++ b/tests/extmod/random_basic.py
@@ -0,0 +1,29 @@
+try:
+    import random
+except ImportError:
+    print("SKIP")
+    raise SystemExit
+
+# check getrandbits returns a value within the bit range
+for b in (1, 2, 3, 4, 16, 32):
+    for i in range(50):
+        assert random.getrandbits(b) < (1 << b)
+
+# check that seed(0) gives a non-zero value
+random.seed(0)
+print(random.getrandbits(16) != 0)
+
+# check that PRNG is repeatable
+random.seed(1)
+r = random.getrandbits(16)
+random.seed(1)
+print(random.getrandbits(16) == r)
+
+# check that zero bits works
+print(random.getrandbits(0))
+
+# check that it throws an error for negative bits
+try:
+    random.getrandbits(-1)
+except ValueError:
+    print("ValueError")