def METHOD_NAME(self, X=None, y=None, groups=None):
    test_folds = self._make_test_folds(X, y)
    for i in range(self.n_splits):
        test_mask = np.zeros(self.n_samples, dtype=bool)
        test_mask[test_folds[i]] = True
        yield test_mask