Implement helper class for generating random values

This commit is contained in:
Jack Grigg 2018-06-14 23:27:03 +12:00
parent 07dc43fd90
commit fb4e0d6efe
No known key found for this signature in database
GPG Key ID: 1B8D649257DB0829
1 changed files with 28 additions and 0 deletions

28
tv_rand.py Normal file
View File

@ -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)]