小编Mec*_*ran的帖子

我是 python 中 Pytest 的新手。

我面临一个棘手的场景，我需要使用 Pytest 模块测试退出代码 - exit(1) 和 exit(0) 。下面是python程序：

 def sample_script():
     count_file  = 0
     if count_file == 0:
        print("The count of files is zero")
     exit(1)
     else:
         print("File are present")
     exit(0)

现在我想测试上述程序的退出代码 exit(1) 和 exit(0) 。使用 Pytest 我们如何构建测试代码，以便我们可以测试或资产函数 sample_script 的退出代码？

请帮我。