code: def _check_component_shapes_helper(
      self, model, component_dists, expected_shape
  ):
      # Given a dict of component marginal distributions, check that the dict
      # has the expected structure and that the distributions have the expected
      # shape (and that shape is static if possible).
      for component in model.components:
          component_stddev = component_dists[component].stddev()
          component_shape = self._get_tensor_shape(component_stddev)
          self.assertAllEqual(component_shape, expected_shape)
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.sts import decomposition
  from tensorflow_probability.python.sts.components import local_linear_trend
  from tensorflow_probability.python.sts.components import seasonal
  from tensorflow_probability.python.sts.components import sum
  from tensorflow_probability.python.sts.forecast import forecast
  from tensorflow_probability.python.sts.internal import missing_values_util
label_desc: Use image node_id 5 for calling the _DecompositionTest obj's underlying member method code with example usage: obj._check_component_shapes_helper(model, component_dists, expected_shape) without return types