def test_fn_layer_weights_state(self):
layer = Fn('2in2out',
lambda x, y: (x + y, np.concatenate([x, y], axis=0)),
n_out=2)
input_signature = None
weights, state = layer.new_weights_and_state(input_signature)
self.assertIsNotNone(weights)
self.assertIsNotNone(state)
self.assertEmpty(weights)
self.assertEmpty(state)
