import numpy as np
import pytest

from sktime.annotation.clasp import ClaSPSegmentation
from sktime.datasets import load_gun_point_segmentation
from sktime.tests.test_switch import run_test_for_class

# load the test dataset
ts, period_size, cps = load_gun_point_segmentation()

# compute a ClaSP segmentation
clasp = ClaSPSegmentation(period_size, n_cps=1)
clasp.fit(ts)
clasp.predict_segments(ts)