code: def testInvalidAxis(self):
      # Test case for GitHub issue 22793.
      with self.cached_session():
          ones = array_ops.ones(shape=[2, 3])
          with self.assertRaises(errors_impl.InvalidArgumentError):
              nn_ops.softmax(ones, axis=2).eval()
  
imports: import unittest
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.platform import test
  from tensorflow.python.platform import tf_logging
label_desc: Use image node_id 24 for calling the SoftmaxTest obj's underlying member method code with example usage: obj.testInvalidAxis() without return types