code: def test_op_reduce_argmax(input_data, axis, device_id, precision):
      if isinstance(axis, list):
          # following numpy, argmax over multiple axis is not supported
          return
      dt = PRECISION_TO_TYPE[precision]
      data = AA(input_data, dtype=dt)
  
      # numpy argmax doesn't support keepdims
      arg_shape = np.amax(data, axis=axis, keepdims=True).shape
      expected_forward = [np.argmax(data, axis=axis).reshape(arg_shape)]
  
      from .. import argmax
  
      _test_unary_op(
          precision,
          device_id,
          argmax,
          input_data,
          expected_forward,
          None,
          {"axis": axis},
      )
  
imports: from __future__ import division
  import numpy
  import itertools
  import pytest
  import cntk
  from .ops_test_utils import unittest_helper, _test_unary_op, AA, precision, PRECISION_TO_TYPE, constant
  from cntk.internal import sanitize_dtype_cntk
  from ..None import reduce_sum, reduce_max, reduce_min, reduce_mean, reduce_log_sum_exp, reduce_prod
label_desc: Use image node_id 11 for calling a global function with example usage: test_op_reduce_argmax(input_data, axis, device_id, precision) without return types