def test_inf_loop(self):
def model(x):
return x
import theano.tensor as T
x = T.tensor4('x')
test_attack = Attack(model, back='th', sess=None)
with self.assertRaises(Exception) as context:
test_attack.generate(x)
self.assertTrue(context.exception)
def test_inf_loop(self):
def model(x):
return x
import numpy as np
import tensorflow as tf
sess = tf.Session()
x = tf.placeholder(tf.float32, shape=(1, ))
test_attack = Attack(model, back='tf', sess=sess)
adv_x = test_attack.generate(x)
with self.assertRaises(Exception) as context:
sess.run(adv_x, feed_dict={x: np.asarray(1.0).reshape((1, ))})
self.assertTrue(context.exception)