def test_invalid_args(self): with tf.Session(): # invalid maxval with np.testing.assert_raises(ValueError): random_uniform([3, 3]).eval() # invalid dtype with np.testing.assert_raises(ValueError): random_uniform([3, 3], maxval=10000, dtype=tf.float32).eval()
def test_min_val(self): with tf.Session(): output = secure_random.random_uniform([6], minval=-10000, maxval=0).eval() for out in output: assert out < 0
def test_wrapper(self): with tf.Session(): output = secure_random.random_uniform([3, 3], maxval=10000).eval() np.testing.assert_array_equal(output.shape, [3, 3])