UnitTesting Python脚本

Question

我是Python新手。

这是否可以在不将代码包装在函数/类中的情况下测试Python脚本？

假设我想用UTs来介绍这个脚本https://github.com/aws-samples/aws-glue-samples/blob/master/examples/join_and_relationalize.py

这有可能为它编写一些UT https://docs.python.org/3/library/unittest.html吗？

问题是:我不能在AWS中运行方法/函数，但是只有脚本是该框架的输入点。

https://docs.aws.amazon.com/glue/latest/dg/aws-glue-programming-python.html

Answer 1

可以测试Python而不需要在函数/类中包装代码吗？

您只需创建运行脚本本身的单元测试(例如，使用subprocess，检查它是否具有正确的retval /输出)。

的问题是:我不能在AWS中运行方法/函数，但是只有脚本才是该框架的输入点。

这实际上并不排除编写函数(甚至类)，除非AWS Glue明确禁止这样做(我认为这是不可能的)。

Python文件既是可运行的脚本，也是可导入的库，这是相当常见的。您只需要“打开”脚本入口点：

def a_function():
    print("a function")

def main():
    a_function()

# magic!
if __name__ == '__main__':
    main()

最后两行是神奇的部分:只有当文件作为脚本运行时，__name__才等于"__main__"。

这意味着，如果您运行上面的文件，它将立即打印“一个函数”，但如果您import它，它将不会做任何事情，直到您调用a_function() (或main())。