<reponame>pashew94/decision_tree_estimators<filename>trees/tests/test_common.py
import pytest

from sklearn.utils.estimator_checks import check_estimator

from StrongTrees import StrongTreeEstimator
from StrongTrees import StrongTreeClassifier
from StrongTrees import StrongTreeTransformer


@pytest.mark.parametrize(
    "estimator",
    [StrongTreeEstimator(), StrongTreeTransformer(), StrongTreeClassifier()]
)
def test_all_estimators(estimator):
    return check_estimator(estimator)

import pytest

from sklearn.utils.estimator_checks import check_estimator

from StrongTrees import StrongTreeEstimator
from StrongTrees import StrongTreeClassifier
from StrongTrees import StrongTreeTransformer


@pytest.mark.parametrize(
    "estimator",
    [StrongTreeEstimator(), StrongTreeTransformer(), StrongTreeClassifier()]
)
def test_all_estimators(estimator):
    return check_estimator(estimator)