adding tests for the mondrian method.

Author: mata

Makefile

@@ -1,7 +1,7 @@
 .PHONY: tests doc build
 
 lint:	
-	flake8 . --exclude=doc mapieenv
+	flake8 . --exclude=doc,mapieenv
 
 type-check:
 	mypy mapie

mapie/tests/test_classification.py

@@ -293,7 +293,8 @@
     "naive": 5 / 9,
     "top_k": 1,
     "raps": 1,
-    "raps_randomized": 8/9
+    "raps_randomized": 8/9,
+    "mondrian": 1
 }
 
 X_toy = np.arange(9).reshape(-1, 1)
@@ -476,6 +477,17 @@
         [False, True, True],
         [False, False, True],
     ],
+    "mondrian": [
+        [True, False, False],
+        [True, False, False],
+        [True, True, False],
+        [True, True, True],
+        [True, True, True],
+        [True, True, True],
+        [False, True, True],
+        [False, True, True],
+        [False, False, True],
+    ],
 }
 
 REGULARIZATION_PARAMETERS = [
@@ -876,7 +888,8 @@ def test_toy_dataset_predictions(strategy: str) -> None:
         alpha=0.5,
         include_last_label=args_predict["include_last_label"],
         agg_scores=args_predict["agg_scores"]
-    )
+    ) 
+
     np.testing.assert_allclose(y_ps[:, :, 0], y_toy_mapie[strategy])
     np.testing.assert_allclose(
         classification_coverage_score(y_toy, y_ps[:, :, 0]),