Refactor amax and test_amax

Details:
- Refactor amax and test_amax
- Minor update to test_asum
- Remove extraneous functions from helpers