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])
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  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 ShapeTest obj's underlying member method code with example usage: obj.test_shape() without return types