Add test for checking the argument keys