def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["test_fun"] = [
        _test_amax_with_negative_dim,
        _test_amax_with_positive_dim,
        _test_amax_with_multiple_axes,
        _test_amax_with_empty_dim,
        _test_amax_keepdim,
        _test_amax_not_keepdim,
    ]
    arg_dict["device"] = ["cpu", "cuda"]
    for arg in GenArgList(arg_dict):
        arg[0](test_case, *arg[1:])