python for selenium 数据驱动测试

 # -*- coding:utf-8 -*-
 """
 数据驱动测试，从 csv 文件中读取数据
 """
 from selenium import webdriver
 import csv, sys

 test_executed = 0
 test_passed = 0
 test_failed = 0
 test_status = True

 try:
     driver = webdriver.Firefox()
     driver.get('C:\BMICalculator.html')

     datafile = open(r'c:\data.csv', 'rb')
     reader = csv.reader(datafile)

     test_executed = 0

     for row in reader:
         test_executed += 1
         print 'Test' + str(test_executed)

         heightField = driver.find_element_by_name('heightCMS')
         heightField.clear()
         heightField.send_keys(row[0])

         weightField = driver.find_element_by_name('weightKg')
         weightField.clear()
         weightField.send_keys(row[1])

         calculateButton = driver.find_element_by_id('Calculate')
         calculateButton.click()

         bmiLabel = driver.find_element_by_name('bmi')
         bmiCategoryLabel = driver.find_element_by_name('bmi_category')

         if bmiLabel.get_attribute('value') == row[2]:
             print "PASS, expected value for BMI <" + row[2] + "> actual <" + bmiLabel.get_attribute('value') + ">"
         else:
             print "FAIL, expected value for BMI <" + row[2] + "> actual <" + bmiLabel.get_attribute('value') + ">"
             test_status = False

         if bmiCategoryLabel.get_attribute('value') == row[3]:
             print "PASS, expected value for BMI Category <" + row[3] + "> actual <" + bmiCategoryLabel.get_attribute('value') + ">"
         else:
             print "Fail, expected value for BMI Category <" + row[3] + "> actual <" + bmiCategoryLabel.get_attribute('value') + ">"
             test_status = False

         if test_status == True:
             test_passed = test_passed + 1
         else:
             test_failed = test_failed + 1

 except :
     print "Unexpected error: ", sys.exc_info()[0]
     raise

 finally:
     print "---------------------------------------------------------------"
     print "Total (" + str(test_executed) + ") Tests Executed"
     print "Total (" + str(test_passed) + ") Tests Passed"
     print "Total (" + str(test_failed ) + ") Tests Failed"
     driver.close()
     datafile.close()