python的unittest测试类代码实例

employ.py 
待测试的类 
 class Employee(): 
  def __init__(self,first_name,last_name,salary): 
    self.first_name=first_name 
    self.last_name=last_name 
    self.salary=salary 
  def give_raise(self,default=5000): 
    return int(self.salary)+default 

test_employ.py 
测试类  
# coding=utf-8 
import unittest 
from employ import Employee  
class TestEmploy(unittest.TestCase): 
  def setUp(self): 
    self.people=Employee("ZHU","Fangya",20000) 
    self.salary=[25000,30000] 
  def test_give_default_raise(self): 
    self.assertEqual(self.people.give_raise(),self.salary[0])  
  def test_give_custome_raise(self): 
    self.default=10000 
    self.assertEqual(self.people.give_raise(default=10000),self.salary[1])   
if __name__=="__main__": 
  unittest.main() 

Done:2 of 2 (0.137s) 
C:\Python27\python.exe "C:\Program Files (x86)\JetBrains\PyCharm 4.0.6\helpers\pycharm\utrunner.py" C:\Users\waiwai\PycharmProjects\untitled2\test_employ.py true 
Testing started at 16:03 ... 
 
Process finished with exit code 0