Python3-运行单元测试时输出__main__文件（来自实际程序，而不是单元测试）

Question

运行测试时，如何使我的__main__文件打印输出？我的意思是从该文件打印，而不是对文件打印进行单元测试。

我有这个示例结构（所有文件都在同一目录中）：

main.py ：

import argparse

print('print me?')  # no output
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('name')
    args = parser.parse_args()
    print(args.name)  # no output

other.py ：

def print_me():
    print('ran print_me')

test.py ：

import unittest
import sh
import other


class TestMain(unittest.TestCase):

    def test_main(self):
        print('test_main')  # prints it.
        sh.python3('main.py', 'test123')

    def test_other(self):
        print('test_other')  # prints it.
        other.print_me()

我用python3 -m nose -s或python3 -m unittest运行它，但没有区别，打印不是从main.py输出，只有直接在测试文件上定义的输出。这是我得到的：

user@user:~/python-programs/test_main$ python3 -m nose -s
test_main
.test_other
ran print_me
.
----------------------------------------------------------------------
Ran 2 tests in 0.040s

OK

P.S。当然，如果我在不使用测试的情况下运行main.py，那么它将正常打印（例如，使用python shell /解释器并使用sh调用main.py，就像在单元测试中一样）

Answer 1

sh.python3开始新过程，nose未捕获其输出。您可以重定向输出，从中输出结果：

print(sh.python3('main.py', 'test123'))