Pytest marks added

Signed-off-by: Álvaro Bacca Peña <[email protected]>

Author: c4ts0up

tests/defences/detector/poison/test_clustering_centroid_analysis.py

@@ -54,6 +54,12 @@ def fit_predict(self, x, **kwargs):
         return self.cluster_labels_to_return
 
 
+@pytest.mark.skip_framework(
+    "keras",
+    "kerastf",
+    "pytorch",
+    "non_dl_frameworks",
+)
 class CCAUDTestCaseBase(unittest.TestCase):
 
     def setUp(self):
@@ -98,12 +104,6 @@ def setUp(self):
         )
 
 
-@pytest.mark.skip_framework(
-    "keras",
-    "kerastf",
-    "pytorch",
-    "non_dl_frameworks",
-)
 class TestInitialization(CCAUDTestCaseBase):
     """
     Unit tests for the ClusteringCentroidAnalysis class, focusing on
@@ -235,6 +235,7 @@ def test_init_invalid_layer_non_relu(self):
             )
 
 
+@pytest.mark.framework_agnostic
 class TestEncodeLabels(unittest.TestCase):
     """
     Unit tests for the ClusteringCentroidAnalysis label encoding, needed for proper clustering.
@@ -260,12 +261,6 @@ def test_encode_multi_labels(self):
         self.assertEqual({"A": 0, "B": 1, "C": 2, "D": 3}, reverse_mapping)
 
 
-@pytest.mark.skip_framework(
-    "keras",
-    "kerastf",
-    "pytorch",
-    "non_dl_frameworks",
-)
 class TestCalculateCentroid(CCAUDTestCaseBase):
     """
     Unit tests for the ClusteringCentroidAnalysis centroid calculations, needed for PCD.
@@ -591,12 +586,6 @@ def fit_predict(self, x, **kwargs):
         self.assertEqual(len(cluster_class_mapping), 0)
 
 
-@pytest.mark.skip_framework(
-    "keras",
-    "kerastf",
-    "pytorch",
-    "non_dl_frameworkslearn",
-)
 class TestFeatureExtraction(CCAUDTestCaseBase):
     """Unit tests for the _feature_extraction function."""
 
@@ -1208,6 +1197,12 @@ def mock_misclass_rate(class_label, deviation):
         self.assertLess(np.mean(self.y_train[np.where(is_clean_np == 1)]), 0.2)
 
 
+@pytest.mark.skip_framework(
+    "keras",
+    "kerastf",
+    "pytorch",
+    "non_dl_frameworks",
+)
 class TestEvaluateDefence(unittest.TestCase):
     """
     Unit tests for the evaluate_defence method of the ClusteringCentroidAnalysis class.