code: def test_forward_pass(self):
      # Test forward computation of A**k.
      f = function(
          inputs=[self.A, self.k],
          outputs=[self.result, self.result_check],
      )
      out, out_check = f(range(10), 101)
      assert np.allclose(out, out_check)
  
file_path: pytensor/tests/scan/test_checkpoints.py
imports: import numpy
  import pytest
  from pytensor.compile.function import function
  from pytensor.gradient import grad
  from pytensor.scan.basic import scan
  from pytensor.scan.checkpoints import scan_checkpoints
  from pytensor.tensor.basic import ones_like
  from pytensor.tensor.type import iscalar, vector
label_desc: Use image node_id 2 for calling the TestScanCheckpoint obj's underlying member method code with example usage: obj.test_forward_pass() without return types