From eea163be5a271ab2e6b427a0c31efb5b009f95a3 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Sat, 18 Jul 2015 04:05:44 -0400 Subject: [PATCH] Grab language code directly from language choices. --- mayan/apps/ocr/test_models.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/mayan/apps/ocr/test_models.py b/mayan/apps/ocr/test_models.py index 60be4f65e5..598112e25b 100644 --- a/mayan/apps/ocr/test_models.py +++ b/mayan/apps/ocr/test_models.py @@ -5,6 +5,7 @@ from django.core.files.base import File from django.test import TestCase from documents.models import DocumentType +from documents.settings import setting_language_choices from documents.test_models import ( TEST_DEU_DOCUMENT_PATH, TEST_DOCUMENT_TYPE, TEST_SMALL_DOCUMENT_PATH ) @@ -29,13 +30,19 @@ class GermanOCRSupportTestCase(TestCase): def setUp(self): self.document_type = DocumentType.objects.create(label=TEST_DOCUMENT_TYPE) + # Get corresponding language code for German from the default language + # choices list + language_code = [language for language in setting_language_choices.value if language[1] == 'German'][0][0] + + self.assertEqual('deu', language_code) + with open(TEST_DEU_DOCUMENT_PATH) as file_object: - self.document = self.document_type.new_document(file_object=File(file_object), language='ger') + self.document = self.document_type.new_document(file_object=File(file_object), language=language_code) def tearDown(self): self.document.delete() self.document_type.delete() def test_ocr_language_backends_end(self): - print self.document.pages.first().ocr_content.content - self.assertTrue('Es bietet einen elektronischen Tresor oder Repository für elektronische Dokumente.' in self.document.pages.first().ocr_content.content) + self.assertTrue('Repository für elektronische Dokumente.' in self.document.pages.first().ocr_content.content) + self.assertTrue('Es bietet einen elektronischen Tresor oder' in self.document.pages.first().ocr_content.content)