如何用一个充满测试用例的输入文件测试Python脚本？

Question

我参加了在线评委比赛，我想用一个.in文件测试我的代码，这个文件充满了测试用例来计时我的算法。如何让脚本从这个.in文件中获取输入？在

Answer 1

从文件和/或stdin读取：

import fileinput
for line in fileinput.input():
    process(line)

Answer 2

所以脚本通常从stdin获取测试用例，现在您想使用文件中的测试用例进行测试？在

如果是这样，请在cmd行上使用<重定向操作：

my_script < testcases.in

Answer 3

PyUnit“Python的标准单元测试框架”可能就是您想要的。在

---

做一个小脚本，这样做：

#!/usr/bin/env python
import sys

def main():
    in_file = open('path_to_file')
    for line in in_file:
        sys.stdout.write(line)

if __name__ == "__main__":
    main()

作为

^{pr2}$