使用python实现以下功能，生成随机的验证码函数（随机生成n位验证码），登录函数（包含用户名，密码，验证码）

生成随机的验证码函数：

import random
import string

def generate_code(n):
    """生成n位验证码"""
    code = ''.join(random.choices(string.ascii_uppercase + string.digits, k=n))
    return code

登录函数：

def login(username, password, code):
    """登录函数"""
    # 假设正确的用户名和密码分别为'admin'和'123456'
    if username == 'admin' and password == '123456':
        # 验证码不区分大小写
        if code.lower() == generate_code(len(code)).lower():
            return True
    return False

使用示例：

# 生成6位验证码
code = generate_code(6)
print(code)

# 登录测试
username = input('请输入用户名：')
password = input('请输入密码：')
input_code = input('请输入验证码：')
if login(username, password, input_code):
    print('登录成功')
else:
    print('登录失败')