文件在Python中运行完毕后自动运行函数

Question

我正在创建一个简单的测试模块，并希望在所有测试完成后显示final_output。

我尝试过使用atexit.register ，但这只在解释器终止时运行该函数。 这适用于自动退出的程序（如PyCharm），但在IDLE等其他程序中，它不会显示。

@atexit.register
def final_output():
    print("Out of {0} tests, {1} succeeded and {2} failed".format(
                    config.total_tests,
                    config.number_success,
                    config.number_failed,
                )
            )

我很感激任何建议。

Answer 1

我认为你最好在一个方法中包装你的测试用例并最终调用summary函数。 就像是：

def test():
    results = run_test_cases() 
    show_results(results) 

这适用于所有情况。