def test_segments_default_not_fitted():
segment_set = ContiguousSegmentSet()
with pytest.raises(NotFittedError):
segment_set.transform()
def test_segments_default():
segment_set = ContiguousSegmentSet()
segment_set.fit(_dataset)
segments = segment_set.transform()
assert segments.shape == _dataset.shape
assert len(segments[0]) == 5
