forked from Rust-related/RustPython
30 lines
800 B
Python
30 lines
800 B
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
|
|
|
|
# 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
|