Esempio n. 1
0
    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)