Update acceptance_test_amax.py

Details:
 - Update the expected value of i for strides not equal to 1 as the max is found, instead of when comparing to the actual index