import unittest
from test import get_formatted_name
class NamesTestCase(unittest.TestCase):
#测试test.py
def test_first_last_name(self):
#能够正确的处理像jj这样的姓名吗?
formatted_name =get_formatted_name('janis','joplin')
self.assertEqual(formatted_name,'Janis Joplin')
if __name__ == '__main__':
unittest.main()
解释
if__name__ == ‘main’
if name == ‘main’: unittest.main()的意思就是:
当程序自己运行时,则调用当前程序中名字以“test”开头的函数
当程序被调用运行时,则不运行程序中名字以“test”开头的函数
在本例中,程序是自己运行的,所以会运行test开头的函数,即test_first_last_name(self)
结论
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
[Finished in 141ms]
修改展现“测试不通过”的情况
def get_formatted_name(first,middle,last):
#生成整洁的姓名
full_name = f"{first} {middle} {last}"
return full_name.title()
E
======================================================================
ERROR: test_first_last_name (__main__.NamesTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
File "X:\python_work\test_name_function.py", line 8, in test_first_last_name
formatted_name =get_formatted_name('janis','joplin')
TypeError: get_formatted_name() missing 1 required positional argument: 'last'
----------------------------------------------------------------------
Ran 1 test in 0.000s
FAILED (errors=1)
[Finished in 141ms]
第一行有个字母E,说明有个测试单元导致了错误
ERROR: test_first_last_name (__main__.NamesTestCase)
这个是NamesTestCase 中的test_first_last_name() 导致了错误
为了解决这个错误,可以做如下修改:(当然了,删除之前的操作也行。。)
def get_formatted_name(first,last,middle= ''):
#生成整洁的姓名
if middle:
full_name = f'{first} {middle} {last}'
else:
full_name = f"{first} {last}"
return full_name.title()
让middle成为可选的部分
.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
[Finished in 141ms]
测试用例通过了
下面增加对middle部分的测试
import unittest
from test import get_formatted_name
class NamesTestCase(unittest.TestCase):
#测试test.py
def test_first_last_name(self):
#能够正确的处理像jj这样的姓名吗?
formatted_name =get_formatted_name('janis','joplin')
self.assertEqual(formatted_name,'Janis Joplin')
def test_first_middle_last_name(self):
#Do name like 'Wolfgang Amadeus Mozart work?'
formatted_name = get_formatted_name('wolfgang','mozart','amadeus')
self.assertEqual(formatted_name,'Wolfgang Amadeus Mozart' )
if __name__ == '__main__':
unittest.main()
新的方法名,必须以test_大头,这样才能在
if name == ‘main’:
unittest.main()