Python 推導式在接口自動化里的運用
Python推導式是一種緊湊且強大的構(gòu)造,允許你用一行代碼高效地生成新的數(shù)據(jù)結(jié)構(gòu)(如列表、字典、集合或生成器)基于現(xiàn)有的可迭代對象。推導式結(jié)合了循環(huán)、條件判斷等結(jié)構(gòu),提供了非常靈活的方式來處理數(shù)據(jù)。
列表推導式
列表推導式(List Comprehensions)
列表推導式是最常見的一種,用于從一個可迭代對象創(chuàng)建一個新列表,其基本語法如下:
[expression for item in iterable if condition]
expression 是針對每個item進行計算并最終放入新列表的表達式。
item 是從iterable中取出的單個元素。
iterable 是任何可迭代對象,如列表、元組、集合、字典或生成器。
condition(可選)是一個過濾條件,只有當該條件為真時,相應的項才會被包含在結(jié)果中。
字典推導式
字典推導式(Dictionary Comprehensions)
字典推導式用來快速創(chuàng)建字典,其語法結(jié)構(gòu)如下:
{key_expression: value_expression for item in iterable if condition}
key_expression 和 value_expression 分別對應新字典中每個鍵值對的鍵和值。
其他部分與列表推導式相似。
集合推導式
集合推導式(Set Comprehensions)
集合推導式與列表推導式類似,但結(jié)果是一個集合(無序且元素唯一),語法為:
{expression for item in iterable if condition}
生成器推導式
生成器推導式(Generator Expressions)
生成器推導式與列表推導式非常相似,但是使用圓括號而不是方括號包圍,生成一個生成器對象,它在迭代時按需產(chǎn)生值,節(jié)省內(nèi)存:
(expression for item in iterable if condition)
代碼示例
場景1: 動態(tài)參數(shù)構(gòu)造
# 假設我們需要測試用戶注冊接口,每個用戶有不同的用戶名和郵箱
params_list = [(f"user_{i}", f"user_{i}@example.com") for i in range(1, 4)]
for username, email in params_list:
payload = {"username": username, "email": email}
print(f"Sending request with payload: {payload}")
# 實際使用時這里會是發(fā)送請求的代碼,例如 `requests.post(url, jsnotallow=payload)`
輸出示例:
Sending request with payload: {'username': 'user_1', 'email': 'user_1@example.com'}
Sending request with payload: {'username': 'user_2', 'email': 'user_2@example.com'}
Sending request with payload: {'username': 'user_3', 'email': 'user_3@example.com'}
場景2: 響應數(shù)據(jù)分析
# 假設我們已經(jīng)獲取到了一個包含用戶信息的API響應
import requests
response = requests.get("https://api.example.com/users")
response_json = response.json()
# 提取所有用戶的ID
user_ids = [user['id'] for user in response_json['data']]
print(f"Extracted User IDs: {user_ids}")
輸出示例:
Extracted User IDs: [1, 2, 3, 4, 5]
場景3: 斷言優(yōu)化
# 假設我們要驗證返回的用戶角色集合是否正確
expected_roles = {"admin", "editor"}
response = requests.get("https://api.example.com/roles")
actual_roles = {role['name'] for role in response.json()}
assert expected_roles == actual_roles, f"Expected roles {expected_roles} but got {actual_roles}"
print("Roles assertion passed.")
輸出示例:
Roles assertion passed.
或者如果斷言失敗,則會輸出錯誤信息。
場景4: 結(jié)果篩選與異常檢測
# 模擬一組API響應
responses = [{"status_code": 200}, {"status_code": 404}, {"status_code": 200}]
failed_responses = [response for response in responses if response["status_code"] != 200]
if failed_responses:
print(f"Failed responses detected: {failed_responses}")
else:
print("All requests succeeded.")
輸出示例:
Failed responses detected: [{'status_code': 404}]
場景5: 測試數(shù)據(jù)準備
# 生成用于測試的數(shù)據(jù)
test_data_gen = (
{"id": idx, "name": f"Test_{idx}", "status": "active" if idx % 2 == 0 else "inactive"}
for idx in range(1, 6)
)
for data in test_data_gen:
print(f"Generated test data: {data}")
# 在實際使用中,這里會是使用這些數(shù)據(jù)進行API請求的代碼
輸出示例:
Generated test data: {'id': 1, 'name': 'Test_1', 'status': 'inactive'}
Generated test data: {'id': 2, 'name': 'Test_2', 'status': 'active'}
Generated test data: {'id': 3, 'name': 'Test_3', 'status': 'inactive'}
Generated test data: {'id': 4, 'name': 'Test_4', 'status': 'active'}
Generated test data: {'id': 5, 'name': 'Test_5', 'status': 'inactive'}