code: def test_norm_axis(shape, axis):
      def fun(x):
          return np.linalg.norm(x, axis=axis)
  
      arr = npr.randn(*shape)
      check_grads(fun, modes=["fwd", "rev"])(arr)
  
imports: from __future__ import absolute_import
  from builtins import range
  import itertools
  import autograd.numpy
  import autograd.numpy.random
  from autograd.test_util import check_grads
  from autograd import tuple
  from autograd import grad
  from functools import partial
  import pytest
label_desc: Use image node_id 19 for calling a global function with example usage: test_norm_axis(shape, axis) and returns: np