关于测试:Python函数默认参数避坑指南

列表是一种常常应用的数据类型。在函数的定义中，经常会应用列表作为参数。

比方，要测试一个接口的数据，接口返回的数据格式如下：

{
  "code": "20000", 
  "data": ["孙悟空","李白","甄姬"], 
  "msg": "success", 
  "status": 0
}

要测试的内容是：返回的 data 数据是否跟需要合乎。在测试之前，须要对数据进一步解决，比方要减少 “ 王昭君 ” 这个元素进去，须要写一个函数：

def add_element(data=["孙悟空","李白","甄姬"]):
    data.append('王昭君')
    return data

print(add_element())
print(add_element())
print(add_element())

在函数定义的时候常常会给参数设置默认值，在这个例子中，将 data 参数设置了默认值，函数定义当前，前面会被频繁的调用，期望值应该是打印如下：

["孙悟空","李白","甄姬","王昭君"]
["孙悟空","李白","甄姬","王昭君"]
["孙悟空","李白","甄姬","王昭君"]

理论后果为：

["孙悟空","李白","甄姬","王昭君"]
["孙悟空","李白","甄姬","王昭君","王昭君"]
["孙悟空","李白","甄姬","王昭君","王昭君","王昭君"]

起因

当定义函数时，会保留函数中默认参数 data 的值，也就是 ["孙悟空","李白","甄姬"], 在每次调用的时候如果传递了新的实参，则应用传递的参数；没有传递，应用定义函数时保留的默认参数。

下面两次调用中，都没有传递新的实参，程序会调用定义函数时保留的默认参数，因为 append() , 在第一次调用当前，默认参数曾经由 ["孙悟空","李白","甄姬"] 扭转为 ["孙悟空","李白","甄姬","王昭君"], 再次执行 append() 之后，就变成了 ["孙悟空","李白","甄姬","王昭君","王昭君"]；同理，第三次又扭转了。

能够应用 id() 函数来定位问题：

def add_element(data=["孙悟空","李白","甄姬"]):
    # id 来示意是不是同一个对象
    print(id(data))
    data.append('王昭君')
    return data

print(add_element())
print(add_element())
print(add_element())

打印进去的 id(data) 为同一个对象，也就是默认参数。如果咱们扭转 第二个 print(add_element()) 为 print(add_element(["孙悟空","李白","甄姬"]))，那么第 2 个 id(data) 就会发生变化，因为它不在是默认值，而是新传进来的实参，理论后果也将变成：

2543416926792
['孙悟空', '李白', '甄姬', '王昭君']
2543418907848
["孙悟空","李白","甄姬", '王昭君']
2543416926792
['孙悟空', '李白', '甄姬', '王昭君', '王昭君']

改良计划

• 如果参数中有列表，尽量不要用它做默认参数
• 如果应用了列表作为默认参数，函数调用时传入实参，而不是省略
• 能够在函数体中另外定义一个变量接管默认参数

def add_element(data=["孙悟空","李白","甄姬"]):
    if data == ["孙悟空","李白","甄姬"]:
        data = ["孙悟空","李白","甄姬"]
    data.append('王昭君')
    return data

我是九柄，公众号【九柄】，分享软件测试文章、面试、教程材料，欢送来看看。