Fixed imports in tests, added missing WheezyCaptcha import

lepture · Oct 23, 2020 · 45be32f · 45be32f
1 parent c41a7e8
commit 45be32f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/captcha/__init__.py b/captcha/__init__.py
@@ -15,4 +15,4 @@
 __homepage__ = 'https://github.com/lepture/captcha'
 
 from .audio import AudioCaptcha
-from .image import ImageCaptcha
+from .image import ImageCaptcha, WheezyCaptcha
diff --git a/tests/test_audio.py b/tests/test_audio.py
@@ -1,6 +1,6 @@
 # coding: utf-8
 
-from captcha.audio import AudioCaptcha
+from captcha import AudioCaptcha
 
 
 def test_audio_generate():

diff --git a/tests/test_image.py b/tests/test_image.py
@@ -3,7 +3,7 @@
 import sys
 
 if not hasattr(sys, 'pypy_version_info'):
-    from captcha.image import ImageCaptcha, WheezyCaptcha
+    from captcha import ImageCaptcha, WheezyCaptcha
 
     def test_image_generate():
         captcha = ImageCaptcha()