ubuntu16.04安装python3 selenium

# -*- coding: utf-8 -*-

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


def open100Chrome():
    begin = 1
    end = 3
    for flag in range(begin, end):
        browser = webdriver.Chrome()
        # Version 61.0.3163.91 (Official Build) (64-bit)
        # browser = webdriver.Firefox()

        browser.get('http://www.yahoo.com')
        assert 'Yahoo' in browser.title
        elem = browser.find_element_by_name('p')  # Find the search box
        elem.send_keys('seleniumhq' + Keys.RETURN)
        browser.quit()


if __name__ == "__main__":
    open100Chrome()

# -*- coding: utf-8 -*-
import unittest
from selenium import webdriver


class GoogleTestCase(unittest.TestCase):
    def setUp(self):
        # self.browser = webdriver.Firefox()
        self.browser = webdriver.Chrome()
        self.addCleanup(self.browser.quit)

    def testPageTitle(self):
        # self.browser.get('http://www.google.com')
        self.browser.get('http://www.yahoo.com')
        self.assertIn('Yahoo', self.browser.title)


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