写一段python自动化测试脚本

下面是一个简单的 Python 自动化测试脚本示例，用于测试一个 Web 应用程序的登录功能：

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

# 打开浏览器
browser = webdriver.Chrome()

# 进入登录页面
browser.get("http://example.com/login")

# 输入用户名和密码
username = browser.find_element_by_name("username")
password = browser.find_element_by_name("password")
username.send_keys("testuser")
password.send_keys("testpass")

# 提交表单
password.send_keys(Keys.RETURN)

# 检查登录是否成功
assert "Logged in" in browser.page_source

# 关闭浏览器
browser.quit()

这个脚本使用 Selenium WebDriver 来模拟用户在浏览器中的操作，输入用户名和密码，提交表单，然后检查页面是否显示 "Logged in" 的文本。如果成功，脚本就会继续执行；如果失败，就会抛出一个 AssertionError。最后，脚本会关闭浏览器窗口。