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)