input: Write a function to automate the login process for a given website using Selenium in Python. The function should initialize a web driver, navigate to the target website, enter the provided username and password, click the submit button, and print a success message if the login is successful. After the login process, the function should close the web driver.
  
  def automate_login(website_url, username, password):
      from selenium import webdriver
      from selenium.webdriver.common.keys import Keys
      import time
  
      # Initialize web driver
      driver = webdriver.Chrome()
  
      # Navigate to the target web page
      driver.get(website_url)
  
      # Locate username field
      username_field = driver.find_element_by_name('username')
  
      # Enter username
      username_field.send_keys(username)
  
      # Locate password field
      password_field = driver.find_element_by_name('password')
  
      # Enter password
      password_field.send_keys(password)
  
      # Locate submit button
      submit_button = driver.find_element_by_name('submit')
  
      # Click submit button
      submit_button.click()
  
      # Wait for login process to complete
      time.sleep(5)
  
      # Login successful
      print('Login successful!')
  
      # Close the web driver
      driver.close()
  
  assert automate_login('http://www.example.com', 'username', 'password') == None
  assert automate_login('http://www.example.com', 'testuser', 'testpass') == None
output: def automate_login(website_url, username, password):
      from selenium import webdriver
      from selenium.webdriver.common.keys import Keys
      import time
  
      driver = webdriver.Chrome()
      driver.get(website_url)
      username_field = driver.find_element_by_name('username')
      username_field.send_keys(username)
      password_field = driver.find_element_by_name('password')
      password_field.send_keys(password)
      submit_button = driver.find_element_by_name('submit')
      submit_button.click()
      time.sleep(5)
      print('Login successful!')
      driver.close()