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()