模拟如果 __name__ == "__main__"：

Question

I need to write test cases for my project, however I am stuck at one line of code. How can I mock this line if __name__ == "__main__" ? Please provide ways to do this. Any help/advice will be helpful.

Answer 1

Don't. You don't need to.

Instead of writing

if __name__ == '__main__':
    # complicated stuff to test...

write

def main():
    # complicated stuff to test...

if __name__ == '__main__':
    main()

Then you can just test main like any other function, without having to mess with __name__ or try to somehow get the if __name__ == '__main__' block to run twice when you want to run two tests on it.