docstr: Verifies that 'op' produces 'expected' when fed input 'op_input' .
  
  Args:
    op: argmin or argmax operator to test.
    axis: integer axis to reduce across.
    output_type: numpy datatype of the output to produce.
    op_input: numpy input array to use as input to 'op'.
    expected: numpy array representing the expected output of 'op'.
code: def _assertOpOutputMatchesExpected(
      self, op, axis, output_type, op_input, expected
  ):
      """Verifies that 'op' produces 'expected' when fed input 'op_input' .
  
      Args:
        op: argmin or argmax operator to test.
        axis: integer axis to reduce across.
        output_type: numpy datatype of the output to produce.
        op_input: numpy input array to use as input to 'op'.
        expected: numpy array representing the expected output of 'op'.
      """
      with self.session() as session:
          with self.test_scope():
              pinp = array_ops.placeholder(
                  dtypes.as_dtype(op_input.dtype),
                  op_input.shape,
                  name="a",
              )
              output = op(pinp, axis=axis, output_type=output_type)
          result = session.run(output, {pinp: op_input})
          self.assertAllEqual(result, expected)
  
file_path: tensorflow/tensorflow/compiler/tests/argminmax_test.py
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the ArgMinMaxTest obj's underlying member method code with example usage: obj._assertOpOutputMatchesExpected(op, axis, output_type, op_input, expected) without return types