5

假设我有两个 python 脚本A.py和B.py. 我正在寻找一种从 A 内部运行 B 的方法，即：

  1. B 相信它是__main__（以便if __name__=="__main__"B 中的块中的代码将运行）
  2. B 实际上不是__main__（因此它不会覆盖"__main__"sys.modules 中的条目）
  3. B 中引发的异常传播到 A（即，可以用exceptA 中的子句捕获）。
  4. 如果没有捕获这些异常，则会生成正确的回溯引用 B 中的行号。

我尝试了各种技术，但似乎没有一个能满足我的所有要求。

  • 使用子流程模块中的工具意味着 B 中的异常不会传播到 A。
  • execfile("B.py", {})运行 B，但它认为它不是主要的。
  • execfile("B.py", {'__name__': '__main__'})使 B.py 认为它是主要的，但它似乎也搞砸了异常回溯打印，因此回溯引用 A 内的行（即真正的__main__）。
  • 使用imp.load_sourcewith__main__作为名称几乎可以工作，除了它实际上修改了 sys.modules，因此踩踏了现有的值__main__

有什么办法可以得到我想要的吗？

（我这样做的原因是因为我正在对现有库进行一些清理。这个库没有真正的测试套件，只有一组产生特定输出的“示例”脚本。我试图利用这些作为测试以确保我的清理不会影响库执行这些示例的能力,所以我想从我的测试套件中运行每个示例脚本.我希望能够在测试脚本中看到这些脚本的异常所以测试脚本可以报告失败的类型，而不仅仅是在示例脚本引发一些异常时报告通用的 SubprocessError。）

4

2 回答 2

2

您的用例是有道理的，但我仍然认为您最好重构测试以便它们可以在外部运行。

你测试脚本有这样的东西吗？

def test():
    pass

if __name__ == '__main__':
    test()

如果没有，也许您应该将测试转换为调用函数，例如test. 然后，从您的主测试脚本中，您可以：

import test1
test1.test()
import test2
test2.test()

为运行测试提供一个通用接口，测试本身使用该接口。检查中有一大块代码__main__不是一件好事。

对不起，我没有回答你问的问题，但我觉得这是正确的解决方案，不会偏离原始测试代码太远。

于 2012-07-06T07:08:41.267 回答
2

回答我自己的问题，因为结果很有趣，可能对其他人有用：

事实证明我错了：execfile("B.py", {'__name__': '__main__'}毕竟是要走的路。它确实正确地产生了回溯。我看到的行号不正确不是异常，而是警告。这些警告是使用warnings.warn("blah", stacklevel=2). 该stacklevel=2参数应该允许在使用已弃用的事物的地方而不是在警告调用时引发诸如弃用警告之类的事情（请参阅文档）。

但是，似乎 execfile-d 文件不计入此目的的“堆栈级别”，并且出于某种目的是不可见的stacklevel。因此，如果 execfile-d 模块顶层的代码导致堆栈级别为 2 的警告，则警告不会在 execfile-d 源文件的正确行号处引发；相反，它在运行 execfile 的文件的相应行号处引发。

这是不幸的，但我可以忍受它，因为这些只是警告，所以它们不会影响测试的实际性能。（起初我没有注意到只有警告受到行号不匹配的影响，因为测试输出中混杂了很多警告和异常。）

于 2012-07-06T07:24:45.810 回答