python进阶：单元测试——保证代码的可重用性

原创

已于 2024-04-10 21:41:15 修改 · 355 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #单元测试

于 2024-04-10 21:36:16 首次发布

本文讨论了在修改旧代码时，如何通过单元测试（如unittest和doctest）提高重构的效率和信心。作者强调了单元测试在重构过程中的重要性，以及如何通过继承TestCase类和在注释中编写测试来实现测试功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

需要为搁置很久的代码修改功能，你是选择在原有代码上重构，还是重新写一份代码？因为有时候重构的时间，可能要比重写一份的时间要长，各种差错要处理，重构完可能也比较乱。因此，大家对于去修改与重构是比较逃避和排斥的，对自己曾经写的代码也不够有信心。

通过单元测试能给自己信心，同时能在重构过程中快速验证。

很多开源项目都有单元测试，里面有各种功能的测试，通过它能加速项目代码的理解

本文只提供一个视角，详细的使用方法可以参考文档和其它博客。

1. unittest简单使用

通过继承重写TestCase的类，来编写测试代码。

import unittest

class TestStringMethods(unittest.TestCase):

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper