Add tests for supported argument types