def test_rejection(self):
m = 1129
seed0 = [
2108217960,
-1340439062,
476173466,
-681389164,
-1502583120,
1663373136,
2144760032,
-1591917499,
]
with tf.Session():
out0 = secure_random.seeded_random_uniform(
[64, 4500],
seed=seed0,
maxval=m,
dtype=tf.int32,
).eval()
out1 = secure_random.seeded_random_uniform(
[64, 4500],
seed=seed0,
maxval=m,
dtype=tf.int32,
).eval()
np.testing.assert_array_equal(out0, out1)
def test_wrapper(self):
expected = [[6610, 5100, 676], [6111, 9801, 5407], [9678, 7188, 8280]]
with tf.Session():
output = seeded_random_uniform([3, 3], seed=seed,
maxval=10000).eval()
np.testing.assert_array_equal(output, expected)
def test_min_val(self):
expected = [[3220, 200, -8648], [2223, 9603, 815], [9356, 4377, 6561]]
with tf.Session():
output = secure_random.seeded_random_uniform([3, 3],
seed=SEED,
minval=-10000,
maxval=10000).eval()
np.testing.assert_array_equal(output, expected)
def test_seed_generation(self):
with tf.Session():
s = secure_random.seed()
minval = -2000
maxval = 0
shape = [2, 3]
output = secure_random.seeded_random_uniform(shape,
seed=s,
minval=minval,
maxval=maxval).eval()
np.testing.assert_array_equal(output.shape, shape)
class TestSeed(unittest.TestCase):
with tf.Session():
s = seed_gen()
minval = -2000
maxval = 0
shape = [2, 3]
output = seeded_random_uniform(shape,
seed=s,
minval=minval,
maxval=maxval).eval()
np.testing.assert_array_equal(output.shape, shape)
def test_invalid_args(self):
with tf.Session():
# invalid seed
with np.testing.assert_raises(ValueError):
seeded_random_uniform([3, 3], maxval=10000, seed=[1]).eval()
# invalid maxval
with np.testing.assert_raises(ValueError):
seeded_random_uniform([3, 3]).eval()
# invalid dtype
with np.testing.assert_raises(ValueError):
seeded_random_uniform([3, 3],
seed=seed,
maxval=10000,
dtype=tf.float32).eval()