python进阶(8)

一、测试函数(unittest)

1、源文件方法（name_function.py）：

def get_formatted_name(first,last):

    """生成全名"""

    full_name=f"{first} {last}"

    return  full_name.title()

2、通过unittest测试（test_name_function.py）

import unittest

from name_function import get_formatted_name
class NamesTestCase( unittest.TestCase ):

    """测试name_function.py"""

    def test_first_last_name(self):

        """能够正确的处理Janis Joplin这样的姓名"""

        formatted_name=get_formatted_name('janis','joplin')

        self.assertEqual( formatted_name, 'Janis Joplin' )
if __name__ == '__main__':

    unittest.main()

查看运行结果：

Ran 1 test in 0.003s
OK

二、测试类(unittest)

1、源文件类（survey.py）

class AnonymousSurvey:

    """收集匿名蒂阿查问卷的答案"""

    def __init__(self,question):

        """存储一个问题，并为存储答案做准备"""

        self.question=question

        self.responses=[]
def show_question(self):

        """显示调查问卷"""

        print(self.question)
def store_response(self,new_response):

        """存储单份调查问卷"""

        self.responses.append(new_response)
def show_results(self):

        """显示收集到的手机答案"""

        print("Survey results")

        for response in self.responses:

            print(f" - {response}")

2、通过unittest测试（test）

import unittest

from survey import AnonymousSurvey
class TestAnonymousSurvey( unittest.TestCase ):

    """针对AnonymousSurvey类的测试"""

    def test_store_single_response(self):

        question="What language did you first learn to speak?"

        my_survey=AnonymousSurvey(question)

        my_survey.store_response('English')

        self.assertIn( 'English', my_survey.responses )
def test_store_three_response(self):

        """测试三个答案并存储正确"""

        question="What language did you first learn to speak?"

        my_survey=AnonymousSurvey(question)

        responses=['English','Spanish','Chinese']

        for response in responses:

            my_survey.store_response(response)

        for response in responses:

            self.assertIn( response, my_survey.responses )
if __name__ == '__main__':

    unittest.main()

查看结果：

Ran 2 tests in 0.002s
OK

三、setp()与tearDown()方法：初始化与清理程序

import unittest

from survey import AnonymousSurvey
class TestAnonymousSurvey( unittest.TestCase ):

    def setUp(self):

        question = "What language did you first learn to speak?"

        self.my_survey=AnonymousSurvey(question)

        self.responses = ['English', 'Spanish', 'Chinese']
def tearDown(self):

        self.responses.clear()
def test_store_single_response(self):

        """针对AnonymousSurvey类的测试"""

        self.my_survey.store_response( self.responses[0] )

        self.assertIn( self.responses[0], self.my_survey.responses )
def test_store_three_response(self):

        """测试三个答案并存储正确"""

        for response in self.responses:

            self.my_survey.store_response( response )

        for response in self.responses:

            self.assertIn( response, self.my_survey.responses )
if __name__ == '__main__':

    unittest.main()

查看运行结果：

Ran 2 tests in 0.001s
OK