code: def test_expand_enumerate_support():
      probs = torch.ones(3, 6) / 6
      d = dist.Categorical(probs)
      actual_enum_shape = (
          TorchDistribution.expand(d, (4, 3))
          .enumerate_support(expand=True)
          .shape
      )
      assert actual_enum_shape == (6, 4, 3)
  
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 25 for calling a global function with example usage: test_expand_enumerate_support() without return types