code: def test_scan(self):
      def cell(carry, x):
          return jn.array([2]) * carry * x, jn.array([3]) * carry * x
  
      carry = jn.array([8.0, 8.0])
      output = jn.array([[3.0, 3.0], [6.0, 6.0], [12.0, 12.0]])
      test_carry, test_output = objax.functional.scan(
          cell, jn.ones((2,)), jn.ones((3,))
      )
      self.assertTrue(jn.array_equal(carry, test_carry))
      self.assertTrue(jn.array_equal(output, test_output))
  
imports: import unittest
  import jax.numpy
  import objax
label_desc: Use image node_id 1 for calling the TestScan obj's underlying member method code with example usage: obj.test_scan() and returns: unknown,  unknown