code: def test_enumerate_support(discrete_dist):
      expected_support = discrete_dist.expected_support
      expected_support_non_vec = discrete_dist.expected_support_non_vec
      if not expected_support:
          pytest.skip("enumerate_support not tested for distribution")
      Dist = discrete_dist.pyro_dist
      actual_support_non_vec = Dist(
          **discrete_dist.get_dist_params(0)
      ).enumerate_support()
      actual_support = Dist(
          **discrete_dist.get_dist_params(-1)
      ).enumerate_support()
      assert_equal(actual_support.data, torch.tensor(expected_support))
      assert_equal(
          actual_support_non_vec.data,
          torch.tensor(expected_support_non_vec),
      )
  
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 15 for calling a global function with example usage: test_enumerate_support(discrete_dist) without return types