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)