Update unit_test_amax

Details:
 - Add unit tests to check different positions and negative values