#栗子

from selenium import webdriverfrom selenium.webdriver.common.by import Byfrom selenium.webdriver.support.ui import Selectfrom selenium.common.exceptions import NoSuchElementExceptionfrom selenium.common.exceptions import NoAlertPresentExceptionimport unittest ,time,re

class BaiduTest(unittest.TestCase):    def setUp(self):        self.driver = webdriver.Chrome()        self.driver.implicitly_wait(30)        self.base_url = 'http://www.baidu.com/'        self.verificationErrors = []        self.accept_next_alert = True

    def test_baidu(self):        driver = self.driver        driver.get(self.base_url,'/')        driver.find_element(By.ID,'kw').clear()        driver.find_element(By.ID,'kw').send_keys('selenium ide')        driver.find_element(By.ID,'su').click()    def is_element_present(self,how,what):        try:            self.driver.find_element(by=how,value=what)        except NoSuchElementException as e:            return False        return True

    def is_alert_present(self):        try:            self.driver.switch_to_alert()        except NoAlertPresentException as e:            return False        return True    def close_alert_and_get_its_text(self):        try:            alert = self.driver.switch_to_alert()            alert_text = alert.text            if self.accept_next_alert:                alert.accept()            else:                alert.dismiss()            return alert_text        finally:            #表示是否继续接受下一个警告，初始化状态为 Ture            self.accept_next_alert = True

    def tearDown(self):        self.driver.quit()        self.assertEqual([],self.verificationErrors)if __name__ == '__main__':    unittest.main()

#栗子 

#用于判断素数def is_prime(n):    if n<=1:        return False    for i in range(2,n):        if n%i == 0:            return False    return True

import unittest

class Test(unittest.TestCase):    def setUp(self):        pass

    def test_case(self):        self.prime = is_prime(4)        self.assertTrue(self.prime,msg='Is not prime!!')

    def tearDown(self):        pass

if __name__ == '__main__':    unittest.main()

#结果  AssertionError: False is not true : Is not prime!!

#栗子 登录126 

from selenium import webdriverfrom selenium.webdriver.common.by import Byimport timeimport unittest

class TestLogin(unittest.TestCase):    def setUp(self):        self.driver = webdriver.Chrome()        self.driver.implicitly_wait(5)        self.base_url = 'http://mail.qq.com'        self.verificationErrors = []        self.accept_next_alert = True

    def test_login(self):        driver = self.driver        driver.get(self.base_url)

        #登录        if driver.find_element(By.CSS_SELECTOR, 'iframe#login_frame').is_displayed():            driver.switch_to.default_content()            driver.switch_to.frame(driver.find_element(By.CSS_SELECTOR, 'iframe#login_frame'))            driver.find_element(By.LINK_TEXT, '帐号密码登录').click()            driver.find_element(By.XPATH, ".//*[@id='u']").clear()            driver.find_element(By.XPATH, ".//*[@id='u']").send_keys('981094301@qq.com')            driver.find_element(By.XPATH, ".//*[@id='p']").clear()            driver.find_element(By.XPATH, ".//*[@id='p']").send_keys('9(zndxcpvsWWXX')            driver.find_element(By.CSS_SELECTOR, 'input#login_button').click()            # 获取断言信息进行断言            text = driver.find_element(By.ID, 'useraddr').text            self.assertEqual(text, '981094301@qq.com')            # 退出            time.sleep(2)            driver.find_element(By.LINK_TEXT, '退出').click()

    def tearDown(self):        self.driver.quit()        self.assertEqual([],self.verificationErrors)

if __name__ == '__main__':    unittest.main()【备注】一定要在主函数里执行，我最开始右键的地方就是test_函数，然后控制台就报错说找不到元素。所以，在哪里右键，就单独执行哪里

#栗子