argparse与pytest兼容问题：SystemExit:2错误解决

当你直接运行脚本时，argparse会解析你传入的命令行参数，功能正常；但执行pytest时，你的模块会被导入，此时如果argparse的parse_args()逻辑写在全局作用域或构造函数的默认执行分支中，它会去解析pytest自身的启动参数（而非你测试时传入的width/height），因找不到必填参数而抛出SystemExit:2异常，导致测试失败。

核心思路是让DungeonGenerator类的构造函数直接接收明确的参数，将命令行参数解析逻辑限制在脚本直接运行的场景下，避免测试时触发不必要的参数解析。

代码修改示例 #

原问题代码（错误写法） #

import argparse

class DungeonGenerator:
    def __init__(self):
        # 构造函数内直接解析命令行参数，测试时会触发错误
        parser = argparse.ArgumentParser()
        parser.add_argument("--width", type=int, required=True)
        parser.add_argument("--height", type=int, required=True)
        args = parser.parse_args()
        self.width = args.width
        self.height = args.height

# 直接运行脚本时的逻辑
generator = DungeonGenerator()

修改后的代码（正确写法） #

import argparse

class DungeonGenerator:
    # 构造函数直接接收width和height参数，与命令行解析解耦
    def __init__(self, width: int, height: int):
        self.width = width
        self.height = height

# 仅当脚本直接运行时，才解析命令行参数
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--width", type=int, required=True)
    parser.add_argument("--height", type=int, required=True)
    args = parser.parse_args()
    # 用解析得到的参数实例化类
    generator = DungeonGenerator(args.width, args.height)
    # 后续业务逻辑...

测试用例写法 #

此时测试时可直接传入参数实例化类，完全不会触发argparse的解析逻辑：

from your_module import DungeonGenerator

def test_dungeon_generator_init():
    generator = DungeonGenerator(width=10, height=15)
    assert generator.width == 10
    assert generator.height == 15

如果需要保留“直接运行脚本时无需手动传参给构造函数”的特性，可给构造函数参数添加默认值，仅在未传入参数时才解析命令行：

import argparse

class DungeonGenerator:
    def __init__(self, width: int = None, height: int = None):
        if width is None or height is None:
            # 仅在未传入参数时，才解析命令行
            parser = argparse.ArgumentParser()
            parser.add_argument("--width", type=int, required=True)
            parser.add_argument("--height", type=int, required=True)
            args = parser.parse_args()
            self.width = args.width
            self.height = args.height
        else:
            self.width = width
            self.height = height

if __name__ == "__main__":
    generator = DungeonGenerator()

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