python 如何用 Hypothesis 來自動化單元測試
高質(zhì)量的代碼離不開單元測試，而設(shè)計單元測試的用例往往又比較耗時，而且難以想到一些極端情況，本文講述如何使用 Hypothesis 來自動化單元測試
刷過力扣算法題的同學(xué)都知道，有時候覺得代碼已經(jīng)很完善了，一提交才發(fā)現(xiàn)很多情況沒有考慮到。然后感嘆力扣的單元測試真的牛。
因此，高質(zhì)量的代碼離不開單元測試，如果現(xiàn)在還沒有寫過單元測試，建議先去學(xué)習(xí)以下常用的單元測試庫[1]，只要實踐過，才能感受到本文開頭提到的那些痛點。
Hypothesis 是一個 Python 庫，用于讓單元測試編寫起來更簡單，運行時功能更強(qiáng)大，可以在代碼中查找您不會想到的極端情況。它穩(wěn)定，強(qiáng)大且易于添加到任何現(xiàn)有測試框架中。它的工作原理是讓您編寫斷言每種情況都應(yīng)該正確的測試，而不僅僅是您偶然想到的那些。
Hypothesis 的基礎(chǔ)知識
典型的單元測試需要自己寫一些測試用例，然后編寫測試函數(shù)，通過一段代碼運行它，然后根據(jù)預(yù)期結(jié)果檢查結(jié)果。
Hypothesis 有所不同。它是基于屬性進(jìn)行單元測試。它通過生成與您的規(guī)范匹配的任意數(shù)據(jù)并檢查在這種情況下程序是否仍然有效。如果找到了一個失敗的用例，它將采用該示例并將其測試用例范圍縮減縮減為一定尺寸，然后對其進(jìn)行簡化，直到找到一個仍會導(dǎo)致問題的小得多的示例。然后將其保存，后續(xù)單元測試時仍會使用這些用例。
現(xiàn)在就讓我們看看怎么用吧。
Hypothesis 快速入門
1、安裝
可以通過 pip 安裝，也可以通過源代碼安裝[2]，也可以安裝一些擴(kuò)展[3]，如下：
pip install hypothesis pip install hypothesis[pandas,django]
2、使用
先寫一段代碼，保存在 mycode.py 中，功能是對字符串進(jìn)行特定的編碼和解碼，內(nèi)容如下：
def encode(input_string):
count = 1
prev = ""
lst = []
for character in input_string:
if character != prev:
if prev:
entry = (prev, count)
lst.append(entry)
count = 1
prev = character
else:
count += 1
entry = (character, count)
lst.append(entry)
return lst
def decode(lst):
q = ""
for character, count in lst:
q += character * count
return q對這段代碼進(jìn)行單元測試，往往需要寫很多測試用例，現(xiàn)在我們使用 hypothesis 來自動為我們測試，編寫 test_mycode.py （文件名隨意），內(nèi)容如下：
from hypothesis import given from mycode import decode,encode from hypothesis.strategies import text import unittest class TestEncoding(unittest.TestCase): @given(text()) def test_decode_inverts_encode(self, s): self.assertEqual(decode(encode(s)), s) if __name__ == "__main__": unittest.main()
可以看出，這里并沒有出現(xiàn)具體的測試用例，而是使用來 text 的策略，相當(dāng)于 hypothesis 自動窮舉來可能的情況，也可以看出它很容易可其他測試框架集成，這里是 unittest?，F(xiàn)在來運行一下看看效果：
(py38env) ? tmp python test_mycode.py Falsifying example: test_decode_inverts_encode( self=<__main__.TestEncoding testMethod=test_decode_inverts_encode>, s='', ) E ====================================================================== ERROR: test_decode_inverts_encode (__main__.TestEncoding) ---------------------------------------------------------------------- Traceback (most recent call last): File "test_mycode.py", line 9, in test_decode_inverts_encode def test_decode_inverts_encode(self, s): File "/Users/aaron/py38env/lib/python3.8/site-packages/hypothesis/core.py", line 1162, in wrapped_test raise the_error_hypothesis_found File "test_mycode.py", line 10, in test_decode_inverts_encode self.assertEqual(decode(encode(s)), s) File "/Users/aaron/tmp/mycode.py", line 14, in encode entry = (character, count) UnboundLocalError: local variable 'character' referenced before assignment ---------------------------------------------------------------------- Ran 1 test in 0.048s FAILED (errors=1)
這里測試出當(dāng)字符串為 '' 的時候會拋出 UnboundLocalError 的異常?，F(xiàn)在我們來修復(fù)這個 bug，然后把所有的測試用例 s 給打印出來，看看它用了哪些測試用例。
encode 函數(shù)加入以下代碼：
if not input_string: return []
test_mycode.py 文件打印出測試用例:
@given(text())
def test_decode_inverts_encode(self, s):
print(f"{s=}")
self.assertEqual(decode(encode(s)), s)再次執(zhí)行：
(py38env) ? tmp python test_mycode.py s='' s='1' s='0' s='0' s='0' s='ā' s='\U000cf5e5' s='0' s='' s='0' s='0' s='E' s=")dù'\x18\U0003deb3¤jd" s='\U0005bc37\x07\U000537a1Yà?i?\U000ce9e5\x0b' s='\U0005bc37\U0005bc37\U000537a1Yà?i?\U000ce9e5\x0b' s='\U0005bc37\U000537a1\U000537a1Yà?i?\U000ce9e5\x0b' s='à\U000537a1\U000537a1Yà?i?\U000ce9e5\x0b' s='\U000965e1\x12\x85&\U000f500a??c' s='\n\U0004466c\x86?\x07' s='ê\U00063f1e\x01G\x88' s='úV\n' s='VV\n' s='\U0008debf湆è' s='\U0008debf湆è' s='\U0008debf湆' s='\U0008debf\U0008debf' s='\U0008debf\U0008debfó]?àq\x82#\U00015196\U0001c8beg' s='\U0008debfgó]?àq\x82#\U00015196\U0001c8beg' s='?' s='?' s='?\U00085b9e' s="?8'?\U00057c38ù;\x07\U000a5ea8ò?=\U00091d5b~8?" s='\U000d6497Y>' s='\U000e0f01' s='\U000e0f01?0y￠KN?' s='\U000e0f01?0y￠KN?' s='\U00050a06' s='?\U000b98b3か\U000ba80aá`?-êu\x8c\x903F?"' s='\x8e\U0004612a\x83?' s='\x8e' s='\x8e\x98\U000fb3e0\U0010d2b3\x10\x82\x94D渥' s='￥W' s='p\U000e5a2aE·`ì' s='\U000b80f8\x12\U000c2d54' s='.\U000703de' s='6\U00010ffa\U000f7994\x8e' s='116\U000f7994\x8e' s='1?6\U000f7994\x8e' s='4?6\U000f7994\x8e' s='4\x8e6\U000f7994\x8e' s='0' s='\U0006a564′D\x93ü\x9eb&i\x1c?' s='\U000ceb6f' s='\U000ceb6f\xa0\x08' s='\U000ceb6f\xa0\x08' s='\U000ceb6f?\x08' s='\U000ceb6f?勻\U0007cc15\U000b2aaa×**' s='\U000ceb6f?勻' s='勻?勻' s='J\x14??' s='q)' s='q)' s='q\U00060931' s='q6' s='\U000e3441' s='\U000e3441\U00019958ˉ' s='\x13' s='\U000f34dbk' s='Kp&t?à' s='\n?\x93' s='\n\n\x93' s='\U00019c8d?3\U00056cbd\U000e3b2f\U00058d302' s='\x90=R\x8b?\x03' s='\x9a' s='\U000147e7' s='\U000147e7\x85\U0007a3ef' s='\U000147e7\U00050a070?>' s='\U000a4089\x0eC+Rá\x02\x97\x9cüì?SS\U0006cbc5;?~\x16\x019V?\U000a32fdQ÷\x15' s='Tú?\x19?Z?' s='??' s='\U000cd45a' s='\U000cd45a\U000e15cb?\x08J\ueb3eú?\x07I\x91\x9a\x18\x16?\x80\x1a' s='\x8f}o\x0eq\x0b' s='\x0e}o\x0eq\x0b' s="\U000e05a3&?o[f?\x8büR'í?t\x97íW\x05\U000caea9\U0008fd74\U000e8f1c1?df??\x13" s='\x10\U000e12e2ù\U0006f96ery\U00014baf\x00\x95\U000dbc92é\U00081613μ\U0003b865Z\U0008cc3c' s='ú\U000b561f\x8f?' s='\tà?÷' s='à\x92?ì\U000618fa\x92' s='\U000aaf94\x94\x84\U000cda69\U0005291a\U000a63det?O\x8a>\U000b458bê.\U00086f07\x1a' s='\U0009754e?U_\xa0\x13PQ\x18o\x07\U0006c9c5.á' s='\U00102456' s='3W??' s='\x14\x1c' s='\x14' s='\x14\U00105bcd"\x10?\x99\U000a5032R\U00056c44V&÷>+\U000aaff2??\U000d7570%a!\U00032553′8x^?' s='\x00\U000e2ac4??UrB' s='\x00\U000e2ac4??UrB' s='\x00\U000e2ac4??UrB' s='a\x1aU\x8a?\U000b2fb9\U0005a586' . ---------------------------------------------------------------------- Ran 1 test in 0.180s OK
從執(zhí)行結(jié)果可以看出，'' 首先被測試，其次 hypothesis 使用了大量的極端測試用例，減輕了手寫的負(fù)擔(dān)，大大提升了效率。
雖然 hypothesis 具有自動記憶功能，你仍然可以顯式的指定某個測試用例一直被測試，而且這是推薦的做法，比如我想在每次的測試中都測試 ''，可以這樣寫：
from hypothesis import given, example
from hypothesis.strategies import text
@given(text())
@example("")
def test_decode_inverts_encode(s):
assert decode(encode(s)) == s這一點非常有用，提升了測試代碼的可讀性，可以用來告訴開發(fā)人員或者未來的自己，輸入的字符串必須要考慮 '' 的情形。
此外，執(zhí)行單元測試，不一定要使用 unittest.main()，也可以這樣，是不是很方便：
if __name__ == "__main__": test_decode_inverts_encode()
3、其他策略參考
從哪里開始
以上僅僅是拋磚引玉，hypothesis 還有很多自動化的特性，不再一一列舉，最好的學(xué)習(xí)方法是邊做，邊嘗試。hypothesis 是一個開源項目，有著詳細(xì)的官方文檔[4]，GitHub 倉庫[5]這里都是你開啟自動化測試的好地方：
參考資料
[1]
庫: https://realpython.com/python-testing/
[2]
源代碼安裝: https://github.com/HypothesisWorks/hypothesis/blob/master/CONTRIBUTING.rst
[3]
擴(kuò)展: https://hypothesis.readthedocs.io/en/latest/extras.html
[4]
官方文檔: https://hypothesis.readthedocs.io/en/latest/quickstart.html#running-tests
[5]
GitHub 倉庫: https://github.com/HypothesisWorks/hypothesis/
以上就是python 如何用 Hypothesis 來自動化單元測試的詳細(xì)內(nèi)容，更多關(guān)于python 用 Hypothesis 來自動化單元測試的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
matplotlib如何設(shè)置坐標(biāo)軸刻度的個數(shù)及標(biāo)簽的方法總結(jié)
這里介紹兩種設(shè)置坐標(biāo)軸刻度的方法,一種是利用pyplot提交的api去進(jìn)行設(shè)置,另一種是通過調(diào)用面向?qū)ο蟮腶pi, 即通過matplotlib.axes.Axes去設(shè)置,需要的朋友可以參考下2021-06-06
Python使用20行代碼實現(xiàn)微信聊天機(jī)器人
這篇文章主要介紹了Python使用20行代碼實現(xiàn)微信聊天機(jī)器人,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-06-06
本文主要介紹了使用Python點云生成3D網(wǎng)格，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
這篇文章主要介紹了python3 cmp實現(xiàn)方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
這篇文章主要介紹了如何用Python生成二維碼、解析二維碼，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下2021-03-03
以視頻爬取實例講解Python爬蟲神器Beautiful Soup用法
這篇文章主要以視頻爬取實例來講解Python爬蟲神器Beautiful Soup的用法,Beautiful Soup是一個為Python獲取數(shù)據(jù)而設(shè)計的包,簡潔而強(qiáng)大,需要的朋友可以參考下2016-01-01
如何將yolo格式轉(zhuǎn)化為voc格式:txt轉(zhuǎn)xml(親測有效)
這篇文章主要介紹了如何將yolo格式轉(zhuǎn)化為voc格式:txt轉(zhuǎn)xml,親測有效,可以使用,本文通過圖文并茂的形式給大家介紹的非常詳細(xì),感興趣的朋友參考下吧2023-12-12
Python 人工智能老照片修復(fù)算法學(xué)習(xí)
老舊或者破損的照片如何修復(fù)呢？本文主要介紹了一個非常不錯的照片恢復(fù)開源項目：Bringing-Old-Photos-Back-to-Life。感興趣的小伙伴快來看看呀2021-11-11
SQLite是一種輕型關(guān)系型數(shù)據(jù)庫，常用于嵌入式設(shè)備和移動應(yīng)用中。Python中內(nèi)置了SQLite模塊，可用于連接和操作SQLite數(shù)據(jù)庫。通過Python SQLite模塊，可以方便地創(chuàng)建、查詢和修改數(shù)據(jù)庫中的數(shù)據(jù)，支持事務(wù)處理和數(shù)據(jù)庫操作的原子性保證2023-04-04
這篇文章主要介紹了python?opencv實現(xiàn)影像拼接，主要包括內(nèi)容又垂直影像拼接vconcat和水平影像拼接hconcat以及縱向拼接多個不同圖片，下面詳細(xì)的相關(guān)內(nèi)容,需要的朋友可以參考一下2022-03-03