我有一个很大的遗留Python方法,它包含大约20个return
语句。
该方法不应该返回None
,但它会这样做。它在一个简单的测试用例中是可重复的。
到目前为止,我使用了一个调试器并逐行遍历代码以找到匹配的return语句。
但有更简单的方法吗?
一旦方法返回None,有没有办法引发异常?
当然我需要看到包含return语句的行。
示例:
def big_method(arg1, some_var):
#.... many returns
if arg1:
return some_var # <------
#... many returns
assert not big_method(True, None) is None
上面是一个简单的代码段。结果:
Traceback (most recent call last):
File "/home/modwork_vums_d/src/manyreturns.py", line 8, in <module>
assert not big_method(True, None) is None
AssertionError
以上回溯并没有多大帮助,因为我希望看到内线 big_method()
。在上面的示例中,我想看看我用<------
标记的内容。
我使用PyCharm,但欢迎使用纯Python或其他解决方案。
仅供记录。有一个后续问题试图在PyCharm中启用此功能:PyCharm: Debugging: r(eturn) Continue execution until the current function returns
答案 0 :(得分:13)
pdb有一个r(eturn)
命令可以满足这个需求:
R(E打开) 继续执行,直到当前函数返回。
示例:
> /Users/georgexsh/wasteland/tmp/app.py(6)<module>()
-> assert not big_method(True, None) is None
(Pdb) s
--Call--
> /Users/georgexsh/wasteland/tmp/app.py(1)big_method()
-> def big_method(arg1, some_var):
(Pdb) r
--Return--
> /Users/georgexsh/wasteland/tmp/app.py(3)big_method()->None
-> return some_var
在pdb doc中查看更多详情。