Python︱测试代码

测试函数

单元测试用于核实函数的某个方面没有问题。

def get_formatted_name(first, last, middle=''):
    """生成整洁的姓名"""
    if middle:
        full_name = first + ' ' + middle + ' ' + last
    else:
        full_name = first + ' ' + last
    return full_name.title()

单元测试

import unittest
from name_function import get_formatted_name

class NameTestCase(unittest.TestCase):
    """测试name_function.py"""

    # unittest框架中，main()方法定义函数时使用的是以"test_xxx"或"testxxx"的格式
    def test_first_last_name(self):
        """能够正确处理像Janis Joplin这样的姓名吗"""
        formatted_name = get_formatted_name('Janis', 'Joplin')
        self.assertEqual(formatted_name, 'Janis Joplin')