Add a test case for custom argument parsing