def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["test_fun"] = [
        _test_weightnorm,
        _test_weightnorm_backward,
    ]
    arg_dict["device"] = ["cpu", "cuda"]
    arg_dict["dim"] = [None, -2, -1, 0, 1]
    for arg in GenArgList(arg_dict):
        arg[0](test_case, *arg[1:])