def test_extract_all_shapelets(params, res_desired):
    """Test that the actual results are the expected ones."""
    res_actual = _extract_all_shapelets(**params)

    for arr_actual, arr_desired in zip(res_actual[0],
                                       res_desired['shapelets']):
        np.testing.assert_array_equal(arr_actual, arr_desired)

    for arr_actual, arr_desired in zip(res_actual[1], res_desired['lengths']):
        np.testing.assert_array_equal(arr_actual, arr_desired)

    for arr_actual, arr_desired in zip(res_actual[2],
                                       res_desired['start_idx']):
        np.testing.assert_array_equal(arr_actual, arr_desired)

    for arr_actual, arr_desired in zip(res_actual[3], res_desired['end_idx']):
        np.testing.assert_array_equal(arr_actual, arr_desired)
Test that the actual results are the expected ones.
pyts/transformation/tests/test_shapelet_transform.py