Add tests for mapped arguments with custom functions