Implement helper class for generating random values
This commit is contained in:
parent
07dc43fd90
commit
fb4e0d6efe
|
@ -0,0 +1,28 @@
|
|||
import os
|
||||
import struct
|
||||
|
||||
|
||||
class Rand(object):
|
||||
def __init__(self, random=os.urandom):
|
||||
self._random = random
|
||||
|
||||
def b(self, l):
|
||||
return self._random(l)
|
||||
|
||||
def v(self, l, f):
|
||||
return struct.unpack(f, self.b(l))[0]
|
||||
|
||||
def u8(self):
|
||||
return self.v(1, 'b')
|
||||
|
||||
def u32(self):
|
||||
return self.v(4, '<I')
|
||||
|
||||
def u64(self):
|
||||
return self.v(8, '<Q')
|
||||
|
||||
def bool(self):
|
||||
return self.u8() % 2 > 0
|
||||
|
||||
def a(self, vals):
|
||||
return vals[self.u8() % len(vals)]
|
Loading…
Reference in New Issue