def test_forward_pass_filter_bank(method, sample_input):
    """Test the forward pass of the FilterBankLayer with IIR filtering."""
    n_chans = 8
    sfreq = 100
    layer = FilterBankLayer(
        n_chans=n_chans,
        sfreq=sfreq,
        band_filters=None,
        method=method,
    )

    output = layer(sample_input)
    assert output.shape == (
        sample_input.shape[0], layer.n_bands, n_chans, sample_input.shape[2]
    ), f"Output shape is incorrect for {method} filtering."
Test the forward pass of the FilterBankLayer with IIR filtering.
test/unit_tests/models/test_modules.py