Refactor axpy and test_axpy

Details:
- Refactor axpy and test_axpy
- Minor update to dot and test_dot