Python异常处理是应对程序运行时错误的重要机制,测试驱动开发(TDD)则是先写测试用例再实现功能的开发模式,将两者结合可以有效提升代码的健壮性和可靠性,减少线上异常问题的出现。

Python异常处理基础
Python使用try-except语句捕获和处理异常,基本结构如下:
try:
# 可能出现异常的代码
result = 10 / 0
except ZeroDivisionError as e:
# 处理除零异常
print(f"捕获到异常:{e}")
result = None
finally:
# 无论是否异常都会执行的代码
print("异常捕获流程结束")
常见的内置异常包括ValueError、TypeError、KeyError等,也可以自定义异常类,继承内置的Exception类即可:
class CustomError(Exception):
"""自定义异常类"""
def __init__(self, message):
self.message = message
super().__init__(self.message)
测试驱动开发(TDD)核心流程
TDD的核心遵循红-绿-重构的循环:
- 红:先编写一个失败的测试用例,此时对应功能尚未实现,测试必然会失败
- 绿:编写最小化的功能代码,让测试用例通过
- 重构:优化现有代码,保证测试用例依然通过,提升代码可读性和性能
Python中常用的测试框架是unittest,也可以根据需求选择pytest等第三方框架,下面的案例会使用unittest演示。
异常处理与TDD结合的实战案例
假设我们需要实现一个用户年龄解析函数,要求输入的字符串必须是数字,且年龄在0到120之间,否则抛出对应异常。
第一步:编写失败的测试用例(红阶段)
先定义测试类,编写覆盖正常场景和异常场景的测试用例:
import unittest
def parse_age(age_str):
"""待实现的函数,当前仅占位"""
pass
class TestParseAge(unittest.TestCase):
def test_valid_age(self):
"""测试合法年龄输入"""
self.assertEqual(parse_age("25"), 25)
def test_invalid_type(self):
"""测试非数字字符串输入"""
with self.assertRaises(ValueError):
parse_age("abc")
def test_age_too_low(self):
"""测试年龄过小场景"""
with self.assertRaises(ValueError):
parse_age("-5")
def test_age_too_high(self):
"""测试年龄过大场景"""
with self.assertRaises(ValueError):
parse_age("150")
if __name__ == "__main__":
unittest.main()
此时运行测试,所有测试用例都会失败,因为parse_age函数还没有实现具体逻辑。
第二步:实现最小功能让测试通过(绿阶段)
实现parse_age函数的逻辑,覆盖所有测试用例的场景:
def parse_age(age_str):
"""解析年龄字符串,返回整数年龄,异常场景抛出ValueError"""
try:
age = int(age_str)
except ValueError:
# 输入不是有效数字时抛出异常
raise ValueError("年龄必须是数字字符串")
if age < 0 or age > 120:
raise ValueError("年龄必须在0到120之间")
return age
再次运行测试,所有用例都会通过,此时完成绿阶段。
第三步:重构优化代码(重构阶段)
当前代码逻辑已经满足需求,我们可以把异常信息统一为自定义异常,让异常语义更清晰:
class AgeParseError(ValueError):
"""年龄解析异常"""
pass
def parse_age(age_str):
"""解析年龄字符串,返回整数年龄,异常场景抛出AgeParseError"""
try:
age = int(age_str)
except ValueError:
raise AgeParseError("年龄必须是数字字符串")
if age < 0 or age > 120:
raise AgeParseError("年龄必须在0到120之间")
return age
同时修改测试用例中的异常类型,对应自定义异常:
class TestParseAge(unittest.TestCase):
def test_valid_age(self):
self.assertEqual(parse_age("25"), 25)
def test_invalid_type(self):
with self.assertRaises(AgeParseError):
parse_age("abc")
def test_age_too_low(self):
with self.assertRaises(AgeParseError):
parse_age("-5")
def test_age_too_high(self):
with self.assertRaises(AgeParseError):
parse_age("150")
再次运行测试,所有用例依然通过,重构完成。
结合实践的注意事项
- 编写测试用例时要覆盖所有异常分支,不要遗漏边界场景,比如空字符串、特殊字符输入等
- 自定义异常要明确继承关系,不要直接抛出通用的
Exception,方便调用方区分不同错误类型 - 功能代码实现时不要过度设计,先满足测试用例的需求,后续再逐步优化
- 如果异常处理逻辑复杂,可以拆分为多个小函数,每个函数对应一个测试用例,降低维护成本
在TDD流程中,异常处理逻辑的设计是前置的,先明确会出现哪些异常场景,再写对应的测试用例,最后实现功能,这样能避免写完功能后再补异常处理导致的逻辑漏洞。