51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.test import override_settings
|
|
|
|
from mayan.apps.common.tests.base import BaseTestCase
|
|
from mayan.apps.documents.tests.mixins import DocumentTestMixin
|
|
from mayan.apps.documents.tests.literals import TEST_DEU_DOCUMENT_PATH
|
|
|
|
from .literals import (
|
|
TEST_DOCUMENT_CONTENT, TEST_DOCUMENT_CONTENT_DEU_1,
|
|
TEST_DOCUMENT_CONTENT_DEU_2
|
|
)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=True)
|
|
class DocumentOCRTestCase(DocumentTestMixin, BaseTestCase):
|
|
# PyOCR's leak descriptor in get_available_languages and image_to_string
|
|
# Disable descriptor leak test until fixed in upstream
|
|
_skip_file_descriptor_test = True
|
|
|
|
def test_ocr_language_backends_end(self):
|
|
content = self.test_document.pages.first().content_object.ocr_content.content
|
|
self.assertTrue(TEST_DOCUMENT_CONTENT in content)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=True)
|
|
class GermanOCRSupportTestCase(DocumentTestMixin, BaseTestCase):
|
|
# PyOCR's leak descriptor in get_available_languages and image_to_string
|
|
# Disable descriptor leak test until fixed in upstream
|
|
_skip_file_descriptor_test = True
|
|
|
|
auto_upload_document = False
|
|
|
|
def setUp(self):
|
|
super(GermanOCRSupportTestCase, self).setUp()
|
|
|
|
with open(TEST_DEU_DOCUMENT_PATH, mode='rb') as file_object:
|
|
self.test_document = self.test_document_type.new_document(
|
|
file_object=file_object, language='deu'
|
|
)
|
|
|
|
def test_ocr_language_backends_end(self):
|
|
content = self.test_document.pages.first().content_object.ocr_content.content
|
|
|
|
self.assertTrue(
|
|
TEST_DOCUMENT_CONTENT_DEU_1 in content
|
|
)
|
|
self.assertTrue(
|
|
TEST_DOCUMENT_CONTENT_DEU_2 in content
|
|
)
|