code: def test_sample_shape_smoke(num_nodes, sample_shape, dtype, bp_iters):
      logits = torch.randn(num_nodes, num_nodes, dtype=dtype)
      d = dist.OneOneMatching(logits, bp_iters=bp_iters)
      with xfail_if_not_implemented():
          values = d.sample(sample_shape)
      assert values.shape == sample_shape + (num_nodes,)
      assert d.support.check(values).all()
  
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 3 for calling a global function with example usage: test_sample_shape_smoke(num_nodes, sample_shape, dtype, bp_iters) without return types