def test_generate_filterbank_failure(): with pytest.raises(AssertionError) as excinfo: generate_filterbank(min_freq=2, max_freq=22, last_low_freq=8, low_width=6, low_overlap=4, high_width=10, high_overlap=6, low_bound=0.2) assert ("max freq needs to be exactly the center " "of a filter band " "Close center: 20") == excinfo.value.message with pytest.raises(AssertionError) as excinfo: generate_filterbank(min_freq=2, max_freq=20, last_low_freq=9, low_width=6, low_overlap=4, high_width=10, high_overlap=6, low_bound=0.2) assert ("last low freq " "needs to be exactly the center of a low_width filter band. " "Close center: 8") == excinfo.value.message
def test_generate_filterbank(): filterbands = generate_filterbank(min_freq=2, max_freq=16, last_low_freq=8, low_width=6, low_overlap=4, high_width=10, high_overlap=6, low_bound=0.2) assert np.array_equal([[0.2,5],[1,7],[3,9],[5,11],[7,17],[11,21]], filterbands)