Fix OCR test.
This commit is contained in:
@@ -18,37 +18,10 @@ class DocumentOCRTestCase(TransactionTestCase):
|
||||
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
|
||||
self.document = self.document_type.new_document(file_object=File(file_object))
|
||||
|
||||
def _test_ocr_language_issue_16(self, language, result):
|
||||
"""
|
||||
Reusable OCR test for a specific language
|
||||
"""
|
||||
|
||||
# Clear the document's first page content and switch language
|
||||
self.document.language = language
|
||||
self.document.save()
|
||||
first_page = self.document.pages.first()
|
||||
first_page.content = ''
|
||||
first_page.save()
|
||||
|
||||
# Queue document for OCR
|
||||
self.document.submit_for_ocr()
|
||||
|
||||
# Make sure content was extracted
|
||||
self.assertTrue(result in self.document.pages.first().ocr_content.content)
|
||||
|
||||
def test_ocr_language_backends_end(self):
|
||||
self._test_ocr_language_issue_16('eng', 'Mayan EDMS')
|
||||
|
||||
# TODO: Add documents with for the corresponding language
|
||||
# def test_ocr_language_backends_deu(self):
|
||||
# self._test_ocr_language_issue_16('deu', 'Mayan EDMS')
|
||||
|
||||
# def test_ocr_language_backends_spa(self):
|
||||
# self._test_ocr_language_issue_16('spa', 'Mayan EDMS')
|
||||
|
||||
# def test_ocr_language_backends_rus(self):
|
||||
# self._test_ocr_language_issue_16('rus', '')
|
||||
|
||||
def tearDown(self):
|
||||
self.document.delete()
|
||||
self.document_type.delete()
|
||||
|
||||
def test_ocr_language_backends_end(self):
|
||||
self.document.submit_for_ocr()
|
||||
self.assertTrue('Mayan EDMS Documentation' in self.document.pages.first().ocr_content.content)
|
||||
|
||||
Reference in New Issue
Block a user