def METHOD_NAME(concat_windows_dataset):
    # Create single input without the batch dimension, just (channels, time)
    X, y, _ = concat_windows_dataset[0]
    assert len(X.shape) == 2
    assert isinstance(y, int)
    # Create dummy data augmentation
    factor = 10
    transform = DummyTransform(k=factor, probability=1.)
    # Check that the transformation forward works and outputs an augmented
    # input with the batch dimension
    transformed_X = transform(X)
    assert transformed_X.shape == X.shape
    # Same check, when also passing y
    transformed_X, transformed_y = transform(X, y)
    assert len(transformed_y.shape) == 1
    assert transformed_X.shape == X.shape