Simplify argument test cases (#19593)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19593
Removes a lot of duplication
Reviewed By: dzhulgakov
Differential Revision: D15039887
fbshipit-source-id: e90fe024b84220dd337fdd314d8f7e3620baec28