def testSeededRNG(self):
N = 100
seed = 8675309
numericdtypes = [ak.int64, ak.float64, ak.bool]
for dt in numericdtypes:
# Make sure unseeded runs differ
a = ak.randint(0, 2**32, N, dtype=dt)
b = ak.randint(0, 2**32, N, dtype=dt)
self.assertFalse((a == b).all())
# Make sure seeded results are same
a = ak.randint(0, 2**32, N, dtype=dt, seed=seed)
b = ak.randint(0, 2**32, N, dtype=dt, seed=seed)
self.assertTrue((a == b).all())
# Uniform
self.assertFalse((ak.uniform(N) == ak.uniform(N)).all())
self.assertTrue((ak.uniform(N, seed=seed) == ak.uniform(N, seed=seed)).all())
# Standard Normal
self.assertFalse((ak.standard_normal(N) == ak.standard_normal(N)).all())
self.assertTrue((ak.standard_normal(N, seed=seed) == ak.standard_normal(N, seed=seed)).all())
# Strings (uniformly distributed length)
self.assertFalse((ak.random_strings_uniform(1, 10, N) == ak.random_strings_uniform(1, 10, N)).all())
self.assertTrue((ak.random_strings_uniform(1, 10, N, seed=seed) == ak.random_strings_uniform(1, 10, N, seed=seed)).all())
# Strings (log-normally distributed length)
self.assertFalse((ak.random_strings_lognormal(2, 1, N) == ak.random_strings_lognormal(2, 1, N)).all())
self.assertTrue((ak.random_strings_lognormal(2, 1, N, seed=seed) == ak.random_strings_lognormal(2, 1, N, seed=seed)).all())
def test_standard_normal(self):
pda = ak.standard_normal(100)
self.assertIsInstance(pda, ak.pdarray)
self.assertEqual(100, len(pda))
self.assertEqual(float, pda.dtype)
pda = ak.standard_normal(np.int64(100))
self.assertIsInstance(pda, ak.pdarray)
self.assertEqual(100, len(pda))
self.assertEqual(float, pda.dtype)
pda = ak.standard_normal(np.int64(100), np.int64(1))
self.assertIsInstance(pda, ak.pdarray)
self.assertEqual(100, len(pda))
self.assertEqual(float, pda.dtype)
npda = pda.to_ndarray()
pda = ak.standard_normal(np.int64(100), np.int64(1))
self.assertTrue((npda == pda.to_ndarray()).all())
with self.assertRaises(TypeError) as cm:
ak.standard_normal('100')
self.assertEqual(
'type of argument "size" must be one of (int, int64); got str instead',
cm.exception.args[0])
with self.assertRaises(TypeError) as cm:
ak.standard_normal(100.0)
self.assertEqual(
'type of argument "size" must be one of (int, int64); got float instead',
cm.exception.args[0])
with self.assertRaises(ValueError) as cm:
ak.standard_normal(-1)
self.assertEqual("The size parameter must be > 0",
cm.exception.args[0])
def test_standard_normal(self):
pda = ak.standard_normal(100)
self.assertIsInstance(pda, ak.pdarray)
self.assertEqual(100,len(pda))
self.assertEqual(float,pda.dtype)
with self.assertRaises(TypeError) as cm:
ak.standard_normal('100')
self.assertEqual('type of argument "size" must be int; got str instead',
cm.exception.args[0])
with self.assertRaises(TypeError) as cm:
ak.standard_normal(100.0)
self.assertEqual('type of argument "size" must be int; got float instead',
cm.exception.args[0])
with self.assertRaises(ValueError) as cm:
ak.standard_normal(-1)
self.assertEqual("The size parameter must be > 0",
cm.exception.args[0])