fix: mock generator_holes_mock

scikit-learn-contrib · Oct 23, 2024 · d43a6e2 · d43a6e2
1 parent 40e900a
commit d43a6e2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/benchmark/test_comparator.py b/tests/benchmark/test_comparator.py
@@ -23,6 +23,12 @@ def generator_holes_mock(mocker: MockerFixture) -> _HoleGenerator:
     mock.split.return_value = [
         pd.DataFrame({"A": [False, False, True], "B": [True, False, False]})
     ]
+    mock.n_splits = 1
+    mock.subset = ["A", "B"]
+    mock.ratio_masked = 0.3
+    mock.random_state = 42
+    mock.groups = None
+
     return mock