def test_init_raises(self, FeatureWTALayer, input_layer):
with pytest.raises(ValueError):
FeatureWTALayer(input_layer, pool_size=3)
def layer(self, FeatureWTALayer, input_layer):
return FeatureWTALayer(input_layer, pool_size=2)