UPD: add nan value in test

thibaultcordier · thibaultcordier · commit 6bbb59c4f3b5 · 2024-07-01T10:41:37.000+02:00
diff --git a/mapie/tests/test_classification.py b/mapie/tests/test_classification.py
@@ -1657,7 +1657,7 @@ def test_pred_loof_isnan() -> None:
 @pytest.mark.parametrize("strategy", [*STRATEGIES])
 def test_pipeline_compatibility(strategy: str) -> None:
     """Check that MAPIE works on pipeline based on pandas dataframes"""
-    X = np.random.randint(0, 100, size=100)
+    X = np.concatenate([np.random.randint(0, 100, size=99), [np.nan]])
     X_cat = np.random.choice(["A", "B", "C"], size=X.shape[0])
     X = pd.DataFrame(
         {

Original file line number	Diff line number	Diff line change
`@@ -1657,7 +1657,7 @@ def test_pred_loof_isnan() -> None:`
`1657`	`1657`	`@pytest.mark.parametrize("strategy", [*STRATEGIES])`
`1658`	`1658`	`def test_pipeline_compatibility(strategy: str) -> None:`
`1659`	`1659`	`"""Check that MAPIE works on pipeline based on pandas dataframes"""`
`1660`		`- X = np.random.randint(0, 100, size=100)`
	`1660`	`+ X = np.concatenate([np.random.randint(0, 100, size=99), [np.nan]])`
`1661`	`1661`	`X_cat = np.random.choice(["A", "B", "C"], size=X.shape[0])`
`1662`	`1662`	`X = pd.DataFrame(`
`1663`	`1663`	`{`