3dc324d3f1
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
30 lines
658 B
Python
30 lines
658 B
Python
try:
|
|
import urandom as random
|
|
except ImportError:
|
|
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 it throws an error for zero bits
|
|
try:
|
|
random.getrandbits(0)
|
|
except ValueError:
|
|
print("ValueError")
|