input: Write a function to automate the Selenium login process for a given website. The function should initialize a web driver, navigate to the target website, locate the username and password fields, enter the credentials, submit the form, and print a success message if the login is successful. If the login is unsuccessful, it should print an error message.
  from selenium import webdriver
  from selenium.webdriver.common.keys import Keys
  import time
  
  def selenium_login(url, username_value, password_value):
      # Initialize web driver
      driver = webdriver.Chrome()
      
      # Navigate to the target web page
      driver.get(url)
      
      # Locate username field
      username = driver.find_element_by_name('username')
      
      # Enter username
      username.send_keys(username_value)
      
      # Locate password field
      password = driver.find_element_by_name('password')
      
      # Enter password
      password.send_keys(password_value)
      
      # 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)
      
      # Check if login was successful
      try:
          driver.find_element_by_id('success_message')
          print('Login successful!')
      except:
          print('Login failed!')
      
      driver.close()
  
  assert selenium_login('http://www.example.com', 'correct_username', 'correct_password') == print('Login successful!')
  assert selenium_login('http://www.example.com', 'wrong_username', 'wrong_password') == print('Login failed!')
output: def selenium_login(url, username_value, password_value):
      driver = webdriver.Chrome()
      driver.get(url)
      username = driver.find_element_by_name('username')
      username.send_keys(username_value)
      password = driver.find_element_by_name('password')
      password.send_keys(password_value)
      submit_button = driver.find_element_by_name('submit')
      submit_button.click()
      time.sleep(5)
      try:
          driver.find_element_by_id('success_message')
          print('Login successful!')
      except:
          print('Login failed!')
      driver.close()