def test_binarize_sparse(sparse_clf_dataset, threshold): # noqa: F811 X_np, X = sparse_clf_dataset t_X = cu_binarize(X, threshold=threshold, copy=True) assert type(t_X) == type(X) sk_t_X = sk_binarize(X_np, threshold=threshold, copy=True) assert_allclose(t_X, sk_t_X)
def test_binarize(failure_logger, clf_dataset, threshold): # noqa: F811 X_np, X = clf_dataset t_X = cu_binarize(X, threshold=threshold, copy=True) assert type(t_X) == type(X) sk_t_X = sk_binarize(X_np, threshold=threshold, copy=True) assert_allclose(t_X, sk_t_X)
def test_binarize_sparse(failure_logger, sparse_clf_dataset, # noqa: F811 threshold): X_np, X = sparse_clf_dataset t_X = cu_binarize(X, threshold=threshold, copy=True) # assert type(t_X) == type(X) if cpx.scipy.sparse.issparse(X): assert cpx.scipy.sparse.issparse(t_X) if scipy.sparse.issparse(X): assert scipy.sparse.issparse(t_X) sk_t_X = sk_binarize(X_np, threshold=threshold, copy=True) assert_allclose(t_X, sk_t_X)