using generate with float64 causes trouble

bjtuxck · Feb 6, 2018 · 1cc2720 · 1cc2720
1 parent 1c12f48
commit 1cc2720
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests_tf/test_attacks.py b/tests_tf/test_attacks.py
@@ -173,6 +173,11 @@ def test_generate_np_gives_adversarial_example_l1(self):
     def test_generate_np_gives_adversarial_example_l2(self):
         self.help_generate_np_gives_adversarial_example(2)
 
+    def test_generate_respects_dtype(self):
+        x = tf.placeholder(dtype=tf.float64, shape=(100, 2))
+        x_adv = self.attack.generate(x)
+        self.assertEqual(x_adv.dtype, tf.float64)
+
     def test_targeted_generate_np_gives_adversarial_example(self):
         x_val = np.random.rand(100, 2)
         x_val = np.array(x_val, dtype=np.float32)