绝对有完美的解决方案！Python内置了一个专门处理这种场景的机制，再也不用靠手动开关来控制测试代码了。

核心方法：利用__name__内置变量 #

Python里每个模块都有一个__name__属性，它的取值取决于模块的使用方式：

• 当你直接运行这个模块文件（比如在命令行敲python myModule.py，或者在编辑器里直接运行），模块的__name__会被设为"__main__"
• 当你把这个模块导入到其他文件中（比如import myModule），__name__就会变成模块的名字（也就是"myModule"）

基于这个特性，你只需要把测试代码放到一个条件判断块里，只有当__name__ == "__main__"时才执行：

# myModule.py

# 这里是你的功能模块代码
def calculate_sum(a, b):
    return a + b

# 测试代码区域
if __name__ == "__main__":
    # 只在直接运行模块时才执行下面的测试逻辑
    print("Running module tests...")
    test_result = calculate_sum(2, 3)
    assert test_result == 5, "Sum calculation failed!"
    print("All tests passed successfully!")

为什么这比布尔开关更好？ #

• 完全不需要手动开/关开关，再也不会因为忘记修改run变量导致导入时执行测试代码
• 这是Python社区的标准实践，其他开发者看到这个写法立刻就能理解你的意图
• 测试代码和功能代码分离清晰，模块的复用性不受影响

当你直接运行myModule.py时，测试代码会正常执行；而当你在主文件里import myModule时，这个if块里的内容会被完全跳过，不会干扰主程序的运行。

内容的提问来源于stack exchange，提问作者Tim