Add tests that check our argument parsing logic

- going the arguments provided by the user
- to the relevant function being called.

Added tests because I discovered a bug that the tests would have caught.