Use arguments more idiomatically in test case