# -*- coding: utf-8 -*-
"""Tests for sktime annotators."""

import pandas as pd
import pytest

from sktime.registry import all_estimators
from sktime.utils._testing.estimator_checks import _make_args

ALL_ANNOTATORS = all_estimators(estimator_types="series-annotator", return_names=False)


@pytest.mark.parametrize("Estimator", ALL_ANNOTATORS)
def test_output_type(Estimator):
    """Test annotator output type."""
    estimator = Estimator.create_test_instance()

    args = _make_args(estimator, "fit")
    estimator.fit(*args)
    args = _make_args(estimator, "predict")
    y_pred = estimator.predict(*args)
    assert isinstance(y_pred, pd.Series)

[(10, 'sktime.registry.all_estimators', 'all_estimators', False, 'from sktime.registry import all_estimators\n'), (13, 'pytest.mark.parametrize', 'pytest.mark.parametrize', False, 'import pytest\n'), (18, 'sktime.utils._testing.estimator_checks._make_args', '_make_args', False, 'from sktime.utils._testing.estimator_checks import _make_args\n'), (20, 'sktime.utils._testing.estimator_checks._make_args', '_make_args', False, 'from sktime.utils._testing.estimator_checks import _make_args\n')]