待测类daicelei.py
class Lei:
def __init__(self,xing,ming,nianxin):
self.xing=xing
self.ming=ming
self.nianxin=nianxin
def zhangxin(self,moren=5000):
self.zhangfu=moren
self.nianxin+=self.zhangfu
return self.nianxin
if __name__=='__main__':
zhangsan=Lei('zhang','san',10000)
zhangsan.zhangxin()
print(zhangsan.nianxin)
测试类ceshilei.py
import unittest
from daicelei import Lei
class TestLei(unittest.TestCase):
def setUp(self):
self.zhangsan=Lei('zhang','san',10000)
def test_zhangsan(self):
cc=self.zhangsan.zhangxin(6000)
self.assertEqual(cc,16000)
def test_zhang(self):
dd=self.zhangsan.zhangxin()
self.assertEqual(dd,15000)
unittest.main()
输出
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
------------------
(program exited with code: 0)
请按任意键继续. . .