Update tests for ShadowAttack

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

art/attacks/evasion/shadow_attack.py

@@ -116,7 +116,6 @@ def generate(self, x: np.ndarray, y: Optional[np.ndarray] = None, **kwargs) -> n
         :param y: An array of a single target label.
         :return: An array with the adversarial examples.
         """
-        print(x.shape)
         if x.shape[0] > 1 or y.shape[0] > 1:
             raise ValueError("This attack only accepts a single sample as input.")
 

tests/attacks/evasion/test_shadow_attack.py

@@ -59,7 +59,7 @@ def test_generate(fix_get_mnist_subset, get_image_classifier_list_for_attack):
 
         x_train_mnist_adv = attack.generate(x=x_train_mnist[0:1], y=y_train_mnist[0:1])
 
-        assert np.max(np.abs(x_train_mnist_adv - x_train_mnist[0:1])) == pytest.approx(0.34966960549354553, 0.06)
+        assert np.max(np.abs(x_train_mnist_adv - x_train_mnist[0:1])) == pytest.approx(0.34966960549354553, abs=0.06)
 
 
 @pytest.mark.skipMlFramework("keras", "scikitlearn")