code: def test_enumerate(num_nodes, dtype):
      logits = torch.randn(num_nodes, num_nodes, dtype=dtype)
      d = dist.OneOneMatching(logits)
      values = d.enumerate_support()
      logging.info(
          "destins = {}, suport size = {}".format(
              num_nodes, len(values)
          )
      )
      assert d.support.check(values), "invalid"
      assert len(set(map(_hash, values))) == len(values), "not unique"
  
file_path: pyro/tests/distributions/test_one_one_matching.py
imports: import logging
  import math
  import pytest
  import torch
  import pyro.distributions
  from tests.common import assert_close, assert_equal, xfail_if_not_implemented
label_desc: Use image node_id 2 for calling a global function with example usage: test_enumerate(num_nodes, dtype) without return types