Fix OCR test.

This commit is contained in:
Roberto Rosario
2015-07-08 04:16:59 -04:00
parent 2033f85874
commit 595f64a25c

View File

@@ -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)