/
githubmirror
/
scikit-learn
Обзор
Документация
Войти
/
githubmirror
/
scikit-learn
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
sklearn/utils/tests/test_mask.py
19 строк
537 B
Jérémie du Boisberranger
MAINT Clean-up utils.__init__: move tests into corresponding test files (#28842)
30 апр 2024, 18:27
Не верифицирован
30 апр 2024, 18:27
f61dd6c
Код
Авторство
О чём код?
import pytest from sklearn.utils._mask import safe_mask from sklearn.utils.fixes import CSR_CONTAINERS from sklearn.utils.validation import check_random_state @pytest.mark.parametrize("csr_container", CSR_CONTAINERS) def test_safe_mask(csr_container): random_state = check_random_state(0) X = random_state.rand(5, 4) X_csr = csr_container(X) mask = [False, False, True, True, True] mask = safe_mask(X, mask) assert X[mask].shape[0] == 3 mask = safe_mask(X_csr, mask) assert X_csr[mask].shape[0] == 3