我想找到一種方法來測試我的代碼是否使用 pytest 針對給定錯誤返回正確的錯誤消息我的代碼適用于字典，在某些情況下，某些鍵可能未定義。在下面的示例中，tag變量由用戶提供，在某些情況下，該鍵可以是未定義的。然后我添加了一個 try/except 語句來捕獲這個異常并向用戶返回一條錯誤消息，解釋問題。try: species = self.species[tag]except KeyError: print(f"Error: the species {tag} is not defined")嘗試為此代碼編寫單元測試時出現我的問題。如何檢查我的代碼是否返回適當的錯誤消息？我可以編寫一個自愿提供錯誤的測試，tag然后檢查標準輸出，但如果我決定更改錯誤消息，我將不得不重寫此測試。在“不要重復自己”的方法中似乎不太合適。也許我什至一開始都沒有正確處理錯誤，并且有更好的方法來使用 Python。
1 回答
慕哥6287543
TA貢獻1831條經驗 獲得超10個贊
我相信你應該改變你的代碼結構來實現你想要的。
class Species:
def myMethod(tag):
try:
species = self.species[tag]
except KeyError:
raise YourCustomError(f"Error: the species {tag} is not defined")
然后，您可以編寫 pytest 函數來檢查在使用帶有錯誤標簽的 myMethod 時是否引發了 YourCustomError。請注意，您可以執行以下操作，而不是 YourCustomError：raise KeyError("custom message")
添加回答
舉報
0/150
提交
取消