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'