code: def test_eye_rank_undist(self):
      self.assertTrue(torch.all(dist.eye_rank(3) == torch.eye(3)))
  
file_path: lightly/tests/utils/test_dist.py
imports: import unittest
  from unittest import mock
  import torch
  from pytest import CaptureFixture
  from lightly.utils import dist
label_desc: Use image node_id 1 for calling the TestDist obj's underlying member method code with example usage: obj.test_eye_rank_undist() without return types