从另一个python程序运行python程序(有一定的要求)

假设我有两个python脚本A.py和B.py.我正在寻找一种方法从A中运行B,方式是：

> B认为它是__main__(因此B中的if __name __ ==“__ main__”块中的代码将运行)
> B实际上不是__main__(因此它不会覆盖sys.modules中的“__main__”条目)
> B中引发的异常传播到A(即,可以用A中的except子句捕获).
>如果没有捕获这些异常,则在B中生成引用行号的正确回溯.

我尝试了各种技术,但似乎都没有满足我的所有要求.

>使用子进程模块中的工具意味着B中的异常不会传播到A.
> execfile(“B.py”,{})运行B,但它不认为它是主要的.
> execfile(“B.py”,{‘_ name__’：’_ _ main__’})使得B.py认为它是主要的,但它似乎也搞砸了异常回溯打印,因此回溯指的是A中的行(即,真正的__main__).
>使用imp.load_source和__main__作为名称几乎可以工作,除了它实际修改sys.modules,因此踩踏__main__的现有值

有什么方法可以得到我想要的东西吗？

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

最佳答案

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

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

def test():
    pass

if __name__ == '__main__':
    test()

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

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

提供运行测试的通用接口,测试本身使用.在__main__检查中有一大块代码是不是很好.

很抱歉,我没有回答你提出的问题,但我觉得这是正确的解决方案而不偏离原始测试代码太远.