code: def test_bipartite_matching_op():
      def assert_match(inputs, x, y, threshold, is_ascend=False):
          for dtype in ["float16", "float32", "float64"]:
              inputs = mx.nd.array(inputs, dtype=dtype)
              x = np.array(x, dtype=dtype)
              y = np.array(y, dtype=dtype)
              a, b = mx.nd.contrib.bipartite_matching(
                  inputs, threshold=threshold, is_ascend=is_ascend
              )
              assert_array_equal(
                  a.asnumpy().astype("int64"), x.astype("int64")
              )
              assert_array_equal(
                  b.asnumpy().astype("int64"), y.astype("int64")
              )
  
      assert_match(
          [[0.5, 0.6], [0.1, 0.2], [0.3, 0.4]],
          [1, -1, 0],
          [2, 0],
          1e-12,
          False,
      )
      assert_match(
          [[0.5, 0.6], [0.1, 0.2], [0.3, 0.4]],
          [-1, 0, 1],
          [1, 2],
          100,
          True,
      )
  
file_path: mxnet/tests/python/unittest/test_contrib_operator.py
imports: from __future__ import print_function
  import numpy
  import mxnet
  import random
  import itertools
  from numpy.testing import assert_allclose, assert_array_equal
  from mxnet.test_utils import *
  from common import assert_raises_cudnn_not_satisfied, xfail_when_nonstandard_decimal_separator
  import unittest
label_desc: Use image node_id 3 for calling a global function with example usage: test_bipartite_matching_op() without return types