Function test_nan_reductions

sklearn/utils/tests/test_array_api.py:434–442

Check NaN reductions like _nanmin and _nanmax

(library, X, reduction, expected)

    ],
)
def test_nan_reductions(library, X, reduction, expected):
    """Check NaN reductions like _nanmin and _nanmax"""
    xp = pytest.importorskip(library)

    with config_context(array_api_dispatch=True):
        result = reduction(xp.asarray(X))

    result = move_to(result, xp=numpy, device="cpu")
    assert_allclose(result, expected)


@pytest.mark.parametrize(