def test_dictionary_keys_ucr_dataset_info(dataset):
    """Test that the length of the dictionart is the expected one."""
    keys_expected = ['n_classes', 'n_timestamps', 'test_size', 'train_size',
                     'type']
    dictionary = ucr_dataset_info(dataset)
    if 'train_size' in dictionary.keys():
        assert sorted(list(dictionary.keys())) == keys_expected
    else:
        for key in dictionary.keys():
            assert sorted(list(dictionary[key].keys())) == keys_expected
Test that the length of the dictionart is the expected one.
pyts/datasets/tests/test_ucr.py