def test_register():
class Custom(filterbanks.Filterbank):
def filters(self):
return None
filterbanks.register_filterbank(Custom)
cls = filterbanks.get("Custom")
assert cls == Custom
with pytest.raises(ValueError):
filterbanks.register_filterbank(filterbanks.STFTFB)
def test_get_none():
assert filterbanks.get(None) is None
def test_get_errors(wrong):
with pytest.raises(ValueError):
# Should raise for anything not a Optimizer instance + unknown string
filterbanks.get(wrong)
def test_make_enc_dec(who):
fb_config = {"n_filters": 500, "kernel_size": 16, "stride": 8}
enc, dec = make_enc_dec("free", who_is_pinv=who, **fb_config)
enc, dec = make_enc_dec(FreeFB, who_is_pinv=who, **fb_config)
assert enc.filterbank == filterbanks.get(enc.filterbank)
