scikit-learn
diff --git a/‎sklearn/model_selection/tests/test_split.py
Copy file name to clipboardExpand all lines: sklearn/model_selection/tests/test_split.py
+2Lines changed: 2 additions & 0 deletions b/‎sklearn/model_selection/tests/test_split.py
Copy file name to clipboardExpand all lines: sklearn/model_selection/tests/test_split.py
+2Lines changed: 2 additions & 0 deletions
@@ -353,6 +353,7 @@ def test_stratified_kfold_ratios(k, shuffle, kfold):
     y = np.array([4] * int(0.10 * n_samples) +
                  [0] * int(0.89 * n_samples) +
                  [1] * int(0.01 * n_samples))
+    # ensure perfect stratification with StratifiedGroupKFold
     groups = np.arange(len(y))
     distr = np.bincount(y) / len(y)
 
@@ -376,6 +377,7 @@ def test_stratified_kfold_label_invariance(k, shuffle, kfold):
                  [0] * int(0.89 * n_samples) +
                  [1] * int(0.01 * n_samples))
     X = np.ones(len(y))
+    # ensure perfect stratification with StratifiedGroupKFold
     groups = np.arange(len(y))
 
     def get_splits(y):