Updated test_attacks.py with the newly added Adversarial Library backend

pralab · May 10, 2024 · d6ed58b · d6ed58b
1 parent f9a3aac
commit d6ed58b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/secmlt/tests/test_attacks.py b/src/secmlt/tests/test_attacks.py
@@ -1,4 +1,5 @@
 import pytest
+from secmlt.adv.evasion.advlib_attacks.advlib_pgd import PGDAdvLib
 from secmlt.adv.evasion.base_evasion_attack import BaseEvasionAttack
 from secmlt.adv.evasion.foolbox_attacks.foolbox_pgd import PGDFoolbox
 from secmlt.adv.evasion.perturbation_models import LpPerturbationModels
@@ -24,6 +25,10 @@
             "foolbox",
             PGDFoolbox.get_perturbation_models(),
         ),
+        (
+            "advlib",
+            PGDAdvLib.get_perturbation_models(),
+        ),
         (
             "native",
             PGDNative.get_perturbation_models(),