code: def test_correctness_lambda_shape(self):
      layer = layers.Lambda(lambda x: x**2, output_shape=lambda x: x)
      output = layer(2 * np.ones((2, 3)))
      self.assertAllClose(4 * np.ones((2, 3)), output)
  
      # Test serialization roundtrip
      config = layer.get_config()
      layer = layers.Lambda.from_config(config, safe_mode=False)
      output = layer(2 * np.ones((2, 3)))
      self.assertAllClose(4 * np.ones((2, 3)), output)
  
file_path: keras/keras/layers/core/lambda_layer_test.py
imports: import numpy
  import pytest
  from keras import layers
  from keras import ops
  from keras import testing
label_desc: Use image node_id 3 for calling the LambdaTest obj's underlying member method code with example usage: obj.test_correctness_lambda_shape() without return types