code: def test_read_torch_data():
      with tempfile.TemporaryDirectory() as test_dir:
          save_tensor = torch.tensor([[1, 2, 4], [2, 5, 3]])
          file_name = os.path.join(test_dir, "save_tensor.pt")
          torch.save(save_tensor, file_name)
          read_tensor = internal.utils._read_torch_data(file_name)
          assert torch.equal(save_tensor, read_tensor)
          save_tensor = read_tensor = None
  
file_path: dgl/tests/python/pytorch/graphbolt/utils/test_internal.py
imports: import json
  import os
  import re
  import tempfile
  import dgl.graphbolt.internal
  import numpy
  import pandas
  import pytest
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: test_read_torch_data() without return types