def test_methods_time_series_forest():
    """Test the supported methods of TimeSeriesForest."""
    clf = TimeSeriesForest(n_estimators=7).fit(X_arange, y)
    assert clf.apply(X_arange).shape == (n_samples, 7)
    assert clf.decision_path(X_arange)[0].shape[0] == n_samples
    assert clf.decision_path(X_arange)[1].shape == (8,)
    assert clf.predict(X_arange).shape == (n_samples,)
    assert clf.predict_proba(X_arange).shape == (n_samples, 3)
    assert isinstance(clf.score(X_arange, y), (float, np.floating))
Test the supported methods of TimeSeriesForest.
pyts/classification/tests/test_time_series_forest.py