python/captcha
1
0
Fork 0
mirror of https://github.com/lepture/captcha.git synced 2026-03-13 09:43:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add more tests

Browse Source
This commit is contained in:
Hsiaoming Yang
2024-07-18 22:07:29 +09:00
parent 5864c0ea8f
commit 8ce85d2c07
3 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -21,3 +21,4 @@ venv/
demo.*
.coverage
coverage.xml
tests/demo.*

10
tests/test_audio.py
View File

@@ -1,7 +1,10 @@
# coding: utf-8
import os
from captcha.audio import AudioCaptcha
ROOT = os.path.abspath(os.path.dirname(__file__))
def test_audio_generate():
captcha = AudioCaptcha()
@@ -14,3 +17,10 @@ def test_audio_random():
captcha = AudioCaptcha()
data = captcha.random(4)
assert len(data) == 4
def test_save_audio():
captcha = AudioCaptcha()
filepath = os.path.join(ROOT, 'demo.wav')
captcha.write('1234', filepath)
assert os.path.isfile(filepath)

10
tests/test_image.py
View File

@@ -1,9 +1,19 @@
# coding: utf-8
import os
from captcha.image import ImageCaptcha
ROOT = os.path.abspath(os.path.dirname(__file__))
def test_image_generate():
captcha = ImageCaptcha()
data = captcha.generate('1234')
assert hasattr(data, 'read')
def test_save_image():
captcha = ImageCaptcha()
filepath = os.path.join(ROOT, 'demo.png')
captcha.write('1234', filepath)
assert os.path.isfile(filepath)