code: def test_prefer_static_shape_dynamic(self, shape):
      x = tf.ones([1, 2], dtype=tf.float64)
  
      @tf.function(
          input_signature=[tf.TensorSpec(shape, dtype=x.dtype)]
      )
      def fn(x):
          return tff.utils.get_shape(x)
  
      shape = self.evaluate(fn(x))
  
      self.assertAllEqual(shape, x.shape.as_list())
  
file_path: tf-quant-finance/tf_quant_finance/utils/shape_utils_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 2 for calling the ShapeUtilsTest obj's underlying member method code with example usage: obj.test_prefer_static_shape_dynamic(shape) and returns: tff