Files
RustPython/extra_tests/snippets/stdlib_random.py
Rex Ledesma 4e2e0b41c6 chore: add ruff format --check (#5774)
* chore: add `ruff format --check`

* fix tests
2025-05-12 14:20:01 +09:00

39 lines
1.0 KiB
Python

import random
random.seed(1234)
# random.randint
assert random.randint(1, 11) == 8
# random.shuffle
left = list(range(10))
right = [2, 7, 3, 5, 8, 4, 6, 9, 0, 1]
random.shuffle(left)
assert left == right
# random.choice
assert random.choice(left) == 5
# random.choices
expected = ["red", "green", "red", "black", "black", "red"]
result = random.choices(["red", "black", "green"], [18, 18, 2], k=6)
assert expected == result
# random.sample
sampled = [0, 2, 1]
assert random.sample(list(range(3)), 3) == sampled
# random.randbytes
random.seed(1234)
zero_size_buf = random.randbytes(0)
assert type(zero_size_buf) is bytes
assert len(zero_size_buf) == 0
non_zero_buf = random.randbytes(4)
assert type(non_zero_buf) is bytes
assert list(non_zero_buf) == list(b"\xb9\x7fi\xf7")
# TODO : random.random(), random.uniform(), random.triangular(),
# random.betavariate, random.expovariate, random.gammavariate,
# random.gauss, random.lognormvariate, random.normalvariate,
# random.vonmisesvariate, random.paretovariate, random.weibullvariate