code: def test_get_unique_invidx_low_high_mem():
      srcids = np.array([14, 0, 3, 3, 0, 3, 9, 5, 14, 12])
      dstids = np.array([10, 16, 12, 13, 10, 17, 16, 13, 14, 16])
      unique_nids = np.array([10, 11, 12, 13, 14, 15, 16, 17, 18, 19])
  
      (
          uniques_low,
          idxes_low,
          srcids_low,
          dstids_low,
      ) = _get_unique_invidx(
          srcids,
          dstids,
          unique_nids,
          low_mem=True,
      )
      (
          uniques_high,
          idxes_high,
          srcids_high,
          dstids_high,
      ) = _get_unique_invidx(
          srcids,
          dstids,
          unique_nids,
          low_mem=False,
      )
      assert np.all(
          uniques_low == uniques_high
      ), f"unique is not expected. {uniques_low} != {uniques_high}"
      assert not np.all(
          idxes_low == idxes_high
      ), f"indices is not expected. {idxes_low} == {idxes_high}"
      assert np.all(
          srcids_low == srcids_high
      ), f"srcids is not expected. {srcids_low} != {srcids_high}"
      assert np.all(
          dstids_low == dstids_high
      ), f"dstdis is not expected. {dstids_low} != {dstids_high}"
  
file_path: dgl/tests/tools/test_convert_partition.py
imports: import os
  import tempfile
  import numpy
  import pytest
  import utils
  from convert_partition import _get_unique_invidx
label_desc: Use image node_id 5 for calling a global function with example usage: test_get_unique_invidx_low_high_mem() without return types