code: def test_mode_full(num_destins, dtype):
      num_sources = 2 * num_destins
      logits = torch.randn(num_sources, num_destins, dtype=dtype) * 10
      d = dist.OneTwoMatching(logits)
      values = d.enumerate_support()
      i = d.log_prob(values).max(0).indices.item()
      expected = values[i]
      actual = d.mode()
      assert_equal(actual, expected)
  
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 14 for calling a global function with example usage: test_mode_full(num_destins, dtype) without return types