def test_normalize(clf_dataset, axis, norm, return_norm): # noqa: F811 X_np, X = clf_dataset if return_norm: t_X, t_norms = cu_normalize(X, axis=axis, norm=norm, return_norm=return_norm) sk_t_X, sk_t_norms = sk_normalize(X_np, axis=axis, norm=norm, return_norm=return_norm) assert_allclose(t_norms, sk_t_norms) else: t_X = cu_normalize(X, axis=axis, norm=norm, return_norm=return_norm) sk_t_X = sk_normalize(X_np, axis=axis, norm=norm, return_norm=return_norm) assert type(t_X) == type(X) assert_allclose(t_X, sk_t_X)
def test_normalize_sparse(sparse_clf_dataset, norm): # noqa: F811 X_np, X = sparse_clf_dataset axis = 0 if X.format == 'csc' else 1 t_X = cu_normalize(X, axis=axis, norm=norm) assert type(t_X) == type(X) sk_t_X = sk_normalize(X_np, axis=axis, norm=norm) assert_allclose(t_X, sk_t_X)