def test_set_params(augmented_mock_clf, augmentation, kwargs, random_batch):
    """Asserts that changing the parameters of a classifier instantiated with
    the `AugmentedDataLoader` is possible. Ensures that
    `braindecode.augmentation` is consistent with `Skorch` API.
    """
    augmented_mock_clf.set_params(iterator_train__transforms=augmentation(**kwargs))
    augmented_mock_clf.set_params(train_split=predefined_split(random_batch))
    assert isinstance(
        augmented_mock_clf.train_split, type(predefined_split(random_batch))
    )
Asserts that changing the parameters of a classifier instantiated with
    the `AugmentedDataLoader` is possible. Ensures that
    `braindecode.augmentation` is consistent with `Skorch` API.
    
test/unit_tests/augmentation/test_transforms.py