def test_parse(self):
def model():
return True
import tensorflow as tf
sess = tf.Session()
test_attack = Attack(model, back='tf', sess=sess)
self.assertTrue(test_attack.parse_params({}))
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
x_val = np.zeros((10, 5, 5, 1))
test_attack = Attack(model, back='th', sess=None)
with self.assertRaises(Exception) as context:
test_attack.generate_np(x_val)
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_val = np.zeros((10, 5, 5, 1))
test_attack = Attack(model, back='tf', sess=sess)
with self.assertRaises(Exception) as context:
test_attack.generate_np(x_val)
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)
def test_back(self):
model = Model()
# Exception is thrown when back is not tf or th
with self.assertRaises(Exception) as context:
Attack(model, back='test', sess=None)
self.assertTrue(context.exception)
def test_sess(self):
# Define empty model
def model():
return True
# Test that it is permitted to provide no session
Attack(model, back='tf', sess=None)
def test_model(self):
import theano.tensor as T
# Exception is thrown when model does not have __call__ attribute
with self.assertRaises(Exception) as context:
model = T.matrix('y')
Attack(model, back='th', sess=None)
self.assertTrue(context.exception)
def test_model(self):
sess = tf.Session()
# Exception is thrown when model does not have __call__ attribute
with self.assertRaises(Exception) as context:
model = tf.placeholder(tf.float32, shape=(None, 10))
Attack(model, sess=sess)
self.assertTrue(context.exception)
def test_back(self):
# Define empty model
def model():
return True
# Exception is thrown when back is not tf or th
with self.assertRaises(Exception) as context:
Attack(model, back='test', sess=None)
self.assertTrue(context.exception)
def test_sess(self):
# Define empty model
def model():
return True
# Exception is thrown when session provided with TH
with self.assertRaises(Exception) as context:
Attack(model, back='th', sess=1)
self.assertTrue(context.exception)
def test_parse(self):
sess = tf.Session()
test_attack = Attack(Model('model', 10, {}), sess=sess)
self.assertTrue(test_attack.parse_params({}))
def test_sess(self):
# Test that it is permitted to provide no session
Attack(Model('model', 10, {}), back='tf', sess=None)
def test_sess(self):
# Define empty model
model = Model()
# Test that it is permitted to provide no session
Attack(model, back='tf', sess=None)
def test_parse(self):
sess = tf.Session()
test_attack = Attack(Model(), back='tf', sess=sess)
self.assertTrue(test_attack.parse_params({}))
def test_parse(self):
def model():
return True
test_attack = Attack(model, back='th', sess=None)
self.assertTrue(test_attack.parse_params({}))
def test_sess(self):
# Test that it is permitted to provide no session.
# The session still needs to be created prior to running the attack.
with tf.Session() as sess:
Attack(Model('model', 10, {}), sess=None)
def test_parse(self):
sess = tf.Session()
test_attack = Attack(Model('model', 10, {}), back='tf', sess=sess)
self.assertTrue(test_attack.parse_params({}))
