自动化登录账号密码并截图保存

import time
import unittest

from selenium import webdriver
from selenium.webdriver.common.by import By

class LoginCase(unittest.TestCase):

def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.maximize_window()

    # 定义登录方法
def login(self, username, password):
    # 登录到网站
    self.driver.get('........')
    self.driver.find_element(By.XPATH, '//*[@id="app"]/div/div[1]/div/div/form/div[1]/div/div/input').send_keys(username)
    self.driver.find_element(By.XPATH, '//*[@id="app"]/div/div[1]/div/div/form/div[2]/div/div/input').send_keys(password)
    self.driver.find_element(By.XPATH, '//*[@id="app"]/div/div[1]/div/div/form/div[4]/div/button/span').click()

def test_login_success(self):
    # 正确用户名密码
    self.login('test1', '12345')
    time.sleep(2)
    # 图片截图储存位置
    self.driver.get_screenshot_as_file("F:pycharm+selenium+zidonghua\login_success.png")

def test_login_pwd_error(self):
    # 正确用户名 错误密码
    self.login('test1', '123')
    time.sleep(2)

    self.driver.get_screenshot_as_file("F:pycharm+selenium+zidonghua\pwd_error.png")

def test_login_user_success(self):
    # 错误用户名 正确密码
    self.login('test', '12345')
    time.sleep(2)

    self.driver.get_screenshot_as_file("F:pycharm+selenium+zidonghua\user_sucess.png")

def test_user__null(self):
    # 空用户名 正确密码
    self.login('', '12345')
    time.sleep(2)

    self.driver.get_screenshot_as_file("F:pycharm+selenium+zidonghua\user__null.png")