code: def test_support_shape(dist):
      for idx in range(dist.get_num_test_data()):
          dist_params = dist.get_dist_params(idx)
          d = dist.pyro_dist(**dist_params)
          assert d.support.event_dim == d.event_dim
          x = dist.get_test_data(idx)
          ok = d.support.check(x)
          assert ok.shape == broadcast_shape(
              d.batch_shape, x.shape[: x.dim() - d.event_dim]
          )
          assert ok.all()
  
imports: import numpy
  import pytest
  import torch
  import pyro
  import pyro.distributions
  from pyro.distributions import TorchDistribution
  from pyro.distributions.testing.gof import auto_goodness_of_fit
  from pyro.distributions.util import broadcast_shape
  from tests.common import TEST_FAILURE_RATE, assert_close, assert_equal, xfail_if_not_implemented
label_desc: Use image node_id 2 for calling a global function with example usage: test_support_shape(dist) without return types