code: def test_shape(self):
      vector_value = [0.0, 1.0]
  
      # case: numpy input
      self.assertAllEqual(ps.shape(np.array(vector_value)), [2])
  
      # case: tensor input with static shape
      self.assertAllEqual(ps.shape(tf.constant(vector_value)), [2])
  
      # case: tensor input with dynamic shape
      if not tf.executing_eagerly():
          shape = ps.shape(
              tf1.placeholder_with_default(vector_value, shape=None)
          )
          self.assertAllEqual(self.evaluate(shape), [2])
  
      # case: tensor input with static shape (with
      # TF_FLAG_TF_SHAPE_DEFAULT_INT64=true which is set by an environment
      # variable when this test is run.)
      self.assertEqual(
          ps.shape(tf.constant(vector_value)).dtype, tf.int64
      )
  
file_path: probability/tensorflow_probability/python/internal/prefer_static_shape64_test.py
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the SizeShapeTestInt64 obj's underlying member method code with example usage: obj.test_shape() without return types