From 7bdd20c650f2c6a43f4b04554db2c8935a027f2a Mon Sep 17 00:00:00 2001 From: frankknoll Date: Wed, 15 Mar 2023 10:22:49 +0100 Subject: [PATCH] starting CaptchaReaderTest --- src/CaptchaReader.py | 2 +- src/CaptchaReaderTest.py | 15 +++++++++++++++ src/HowBadIsMyBatch.ipynb | 2 +- src/captcha.ipynb | 2 +- 4 files changed, 18 insertions(+), 3 deletions(-) create mode 100644 src/CaptchaReaderTest.py diff --git a/src/CaptchaReader.py b/src/CaptchaReader.py index fe3df56eb7d..82d803ff21a 100644 --- a/src/CaptchaReader.py +++ b/src/CaptchaReader.py @@ -63,7 +63,7 @@ def _getModel(): return model def load_model(): - model = keras.models.load_model('model') + model = keras.models.load_model('src/model') return keras.models.Model( model.get_layer(name="image").input, model.get_layer(name="dense2").output) diff --git a/src/CaptchaReaderTest.py b/src/CaptchaReaderTest.py new file mode 100644 index 00000000000..b6d46da6f0c --- /dev/null +++ b/src/CaptchaReaderTest.py @@ -0,0 +1,15 @@ +import unittest +from CaptchaReader import getTextInCaptchaImage + + +class CaptchaReaderTest(unittest.TestCase): + + def test_getTextInCaptchaImage(self): + # Given + textInCaptchaImage = '1Ad47a' + + # When + textInCaptchaImageActual = getTextInCaptchaImage(f'src/captchas/VAERS/{textInCaptchaImage}.jpeg') + + # Then + self.assertEqual(textInCaptchaImageActual, textInCaptchaImage) diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb index a1ef9093c6d..d3d29ca3be4 100644 --- a/src/HowBadIsMyBatch.ipynb +++ b/src/HowBadIsMyBatch.ipynb @@ -329,7 +329,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.8" + "version": "3.9.15" }, "vscode": { "interpreter": { diff --git a/src/captcha.ipynb b/src/captcha.ipynb index a25139bc6cc..c0a19e3a785 100644 --- a/src/captcha.ipynb +++ b/src/captcha.ipynb @@ -949,7 +949,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.8" + "version": "3.9.15" } }, "nbformat": 4,