def test_ocr(self):
# Do OCR on easy image to ensure it's working
captcha_solver = CaptchaSolver(None)
test_img1 = cv2.imread(os.path.join(os.getcwd(), 'test_ocr_valid.png'))
test_img2 = cv2.imread(os.path.join(os.getcwd(), 'test_ocr_invalid.png'))
captcha_text1 = captcha_solver.read_captcha(test_img1)
captcha_text2 = captcha_solver.read_captcha(test_img2)
assert captcha_text1 == '123'
assert captcha_text2 == '12'
def test_ocr(self, tmpdir, testdatadir):
# Do OCR on easy image to ensure it's working
td = tmpdir.mkdir('captcha')
captcha_solver = CaptchaSolver(None, outdir=td)
test_img1 = cv2.imread(testdatadir.join('test_ocr_valid.png').strpath)
test_img2 = cv2.imread(
testdatadir.join('test_ocr_invalid.png').strpath)
captcha_text1 = captcha_solver.read_captcha(test_img1)
captcha_text2 = captcha_solver.read_captcha(test_img2)
assert captcha_text1 == '123'
assert captcha_text2 == '12'