code: def test_grad_and_aux():
      A = npr.randn(5, 4)
      x = npr.randn(4)
  
      f = lambda x: (np.sum(np.dot(A, x)), x**2)
      g = lambda x: np.sum(np.dot(A, x))
  
      assert len(grad_and_aux(f)(x)) == 2
  
      check_equivalent(grad_and_aux(f)(x)[0], grad(g)(x))
      check_equivalent(grad_and_aux(f)(x)[1], x**2)
  
imports: from __future__ import absolute_import
  from builtins import range
  import warnings
  from functools import partial
  import autograd.numpy
  import autograd.numpy.random
  from autograd.test_util import check_grads, check_equivalent
  from autograd.tracer import primitive, isbox
  from autograd import grad, elementwise_grad, jacobian, value_and_grad, hessian_tensor_product, hessian, make_hvp, tensor_jacobian_product, checkpoint, make_jvp, make_ggnvp, grad_and_aux
label_desc: Use image node_id 28 for calling a global function with example usage: test_grad_and_aux() without return types