def audio_batch(request, channels_first):
    """
    Audio fixtures of shape `(batch_size, channels, samples)` or `(batch_size, samples, channels)`.
    """
    channels = request.param
    audio_input = AudioInput(channels_first, channels)
    test_input = audio_input.get_data()
    test_output = test_input.copy()
    return test_input, test_output, audio_input.sample_rate

    Audio fixtures of shape `(batch_size, channels, samples)` or `(batch_size, samples, channels)`.
    
tests/defences/preprocessor/test_mp3_compression_pytorch.py