Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
import randomgen as rg
BUFFER_SIZE = 256 * 2 ** 20
DESCRIPTION = """
A driver that simplifies testing bit generators using PractRand.
"""
CONFIG = {
rg.PCG32: {"output": 32, "seed": 64, "seed_size": 64},
rg.PCG64: {"output": 64, "seed": 128, "seed_size": 128},
rg.ThreeFry: {"output": 64, "seed": 256, "seed_size": 64},
rg.Xoshiro256: {"output": 64, "seed": 256, "seed_size": 64},
rg.Philox: {"output": 64, "seed": 256, "seed_size": 64},
rg.SFMT: {"output": 64, "seed": 128, "seed_size": 32},
rg.LXM: {"output": 64, "seed": 128, "seed_size": 32},
rg.SFC64: {"output": 64, "seed": 128, "seed_size": 32},
rg.AESCounter: {"output": 64, "seed": 128, "seed_size": 32},
}
def reorder_bytes(a):
dtype = a.dtype
assert dtype in (np.uint32, np.uint64)
cols = 8 if dtype == np.uint64 else 4
a = a.view(np.uint8).reshape((-1, cols))
return a.ravel("F").view(dtype)
def pack_bits(a, bits):
print("Packing bits")
assert bits > 32
LCG128Mix,
Philox,
Romu,
ThreeFry,
Xoshiro256,
Xoshiro512,
)
ALL_BIT_GENS = [
AESCounter,
ChaCha,
DSFMT,
EFIIX64,
HC128,
JSF,
LXM,
PCG64,
LCG128Mix,
MT19937,
Philox,
SFC64,
SFMT,
SPECK128,
ThreeFry,
Xoshiro256,
Xoshiro512,
Romu,
]
JUMPABLE = [bg for bg in ALL_BIT_GENS if hasattr(bg, "jumped")]
SPECIALS = {
ChaCha: {"rounds": [8, 20]},
REPEAT = 10
SIZE = 25000
PRNGS = [
ChaCha8,
JSF32,
Philox4x32,
ThreeFry2x64,
ThreeFry4x32,
Philox2x64,
DSFMT,
MT64,
MT19937,
PCG64,
PCG64DXSM128,
PCG64DXSM,
LXM,
SFMT,
AESCounter,
ChaCha,
Philox,
ThreeFry,
Xoshiro256,
Xoshiro512,
JSF,
Romu,
RomuTrio,
HC128,
SPECK128,
SFC64,
EFIIX64,
]