code: def test_shift(self, input_arr, expected_output):
      # Without shift, softmax overflows and gets a RuntimeWarning, but just returns nan
      with pytest.warns(RuntimeWarning):
          output_no_shift = softmax(input_arr, shift=False)
      assert np.isnan(output_no_shift).all()
  
      output_shift = softmax(input_arr, shift=True)
      assert np.allclose(output_shift, expected_output)
  
file_path: cleanlab/tests/internal/test_numerics.py
imports: import pytest
  import numpy
  from cleanlab.internal.numerics import softmax
label_desc: Use image node_id 4 for calling the TestSoftmax obj's underlying member method code with example usage: obj.test_shift(input_arr, expected_output) without return types