Beispiel #1
0
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)
Beispiel #2
0
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)