class Employee():
def __init__(self,first_name,last_name,salary):
self.first_name=first_name
self.last_name=last_name
self.salary=salary
def get_raise(self,step=5000):
self.salary+=step
import unittest
from employee import Employee
class TestEmployeeCase(unittest.TestCase):
def setUp(self):
self.m_employee=Employee('Luo','Aa',7000)
def test_give_default_raise(self):
self.m_employee.get_raise()
self.assertEqual(self.m_employee.salary,12000)
def test_give_costom_raise(self):
self.m_employee.get_raise(6000)
self.assertEqual(str(self.m_employee.salary),'13000')
unittest.main()
测试的时候,控制台显示完成0个测试, 原因是unittest.main()不能写在测试类的代码块里。应放在最外面
def test_give_costom_raise(self):
self.m_employee.get_raise(6000)
self.assertEqual(str(self.m_employee.salary),'13000')
unittest.main()