def test_add_dummy_feature_sparse(sparse_clf_dataset, value): # noqa: F811 X_np, X = sparse_clf_dataset t_X = cu_add_dummy_feature(X, value=value) assert type(t_X) == type(X) sk_t_X = sk_add_dummy_feature(X_np, value=value) assert_allclose(t_X, sk_t_X)
def test_add_dummy_feature_sparse(sparse_dataset_with_coo, # noqa: F811 value): X_np, X = sparse_dataset_with_coo t_X = cu_add_dummy_feature(X, value=value) # assert type(t_X) == type(X) if cp.sparse.issparse(X): assert cp.sparse.issparse(t_X) if scipy.sparse.issparse(X): assert scipy.sparse.issparse(t_X) sk_t_X = sk_add_dummy_feature(X_np, value=value) assert_allclose(t_X, sk_t_X)