def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["test_fun"] = [
        _test_cast_float2int,
        _test_cast_int2float,
        _test_cast_bool2int16,
        _test_cast_backward,
        # _test_cast_with_non_contiguous_input,
    ]
    arg_dict["device"] = ["cpu", "cuda"]
    arg_dict["shape"] = [(2, 3), (2, 3, 4), (2, 3, 4, 5)]
    for arg in GenArgList(arg_dict):
        arg[0](test_case, *arg[1:])