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_parse(self): sess = tf.Session() test_attack = Attack(Model('model', 10, {}), sess=sess) self.assertTrue(test_attack.parse_params({}))
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_parse(self): sess = tf.Session() test_attack = Attack(Model('model', 10, {}), back='tf', sess=sess) self.assertTrue(test_attack.parse_params({}))