code: def testStaticStopGradient(self):
      self.assertEqual(extensions.stop_gradient(5.0), 5.0)
      self.assertEqual(type(extensions.stop_gradient(5.0)), type(5.0))
  
      self.assertEqual(
          extensions.stop_gradient(tf_np.asarray(5.0)), 5.0
      )
      self.assertNotEqual(
          type(extensions.stop_gradient(tf_np.asarray(5.0))), type(5.0)
      )
  
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 48 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testStaticStopGradient() without return types