code: def test_take_eigpairs():
      from deepchem.utils.differentiation_utils.symeig import (
          _take_eigpairs,
      )
  
      eival = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
      eivec = torch.tensor(
          [
              [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]],
              [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]],
          ]
      )
      neig = 2
      mode = "lowest"
      eival, eivec = _take_eigpairs(eival, eivec, neig, mode)
      assert torch.allclose(
          eival, torch.tensor([[1.0, 2.0], [4.0, 5.0]])
      )
      assert torch.allclose(
          eivec,
          torch.tensor(
              [
                  [[1.0, 2.0], [4.0, 5.0], [7.0, 8.0]],
                  [[1.0, 2.0], [4.0, 5.0], [7.0, 8.0]],
              ]
          ),
      )
  
imports: import pytest
label_desc: Use image node_id 26 for calling a global function with example usage: test_take_eigpairs() without return types