code: def testMap(self):
      shape = [2, 3]
      dtype = tf_np.int32
      xs1 = tf_np.zeros(shape, dtype)
      xs2 = tf_np.ones(shape, dtype)
      ys_expected = [xs2 + 10, xs1 + 20]
  
      def f(x):
          self.assertIsInstance(x, tuple)
          for a in x:
              self.assertEqual(a.shape, shape[1:])
          x1, x2 = x
          return [x2 + 10, x1 + 20]
  
      ys = extensions.tf_map(f, (xs1, xs2))
      self.assertIsInstance(ys, list)
      self.assertAllClose(ys, ys_expected)
  
imports: import functools
  import itertools
  from absl import flags
  from absl.testing import parameterized
  from jax import lax
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy import extensions
  import trax.tf_numpy.numpy
label_desc: Use image node_id 26 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testMap() and returns: list