code: def test_softmax_correctness_with_axis(self):
      softmax_layer = softmax.Softmax(axis=(1))
      input = np.array([[1.0, 2.0, 1.0], [1.0, 2.0, 1.0]])
      expected_output = np.array(
          [
              [0.21194157, 0.5761169, 0.21194157],
              [0.21194157, 0.5761169, 0.21194157],
          ]
      )
      result = softmax_layer(input)
      self.assertAllClose(result, expected_output)
  
file_path: keras/keras/layers/activations/softmax_test.py
imports: import numpy
  import pytest
  from keras import testing
  from keras.layers.activations import softmax
label_desc: Use image node_id 4 for calling the SoftmaxTest obj's underlying member method code with example usage: obj.test_softmax_correctness_with_axis() without return types