paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

test case fixes.

Browse Source
This commit is contained in:
jonaswinkler 2020-11-27 14:06:37 +01:00
parent ea9de1bcf1
commit 06cfc3113a
2 changed files with 26 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/paperless_tesseract/tests/test_ocr.py
View File

@ -1,34 +0,0 @@
import os
from unittest import mock, skipIf
from django.test import TestCase
from ..parsers import strip_excess_whitespace
class TestOCR(TestCase):
text_cases = [
("simple string", "simple string"),
(
"simple newline\n testing string",
"simple newline\ntesting string"
),
(
"utf-8 строка с пробелами в конце ",
"utf-8 строка с пробелами в конце"
)
]
def test_strip_excess_whitespace(self):
for source, result in self.text_cases:
actual_result = strip_excess_whitespace(source)
self.assertEqual(
result,
actual_result,
"strip_exceess_whitespace({}) != '{}', but '{}'".format(
source,
result,
actual_result
)
)

36
src/paperless_tesseract/tests/test_parser.py
View File

@ -1,6 +1,4 @@
import os
import shutil
import tempfile
import uuid
from typing import ContextManager
from unittest import mock
@ -8,7 +6,8 @@ from unittest import mock
from django.test import TestCase, override_settings
from documents.parsers import ParseError, run_convert
from paperless_tesseract.parsers import RasterisedDocumentParser, get_text_from_pdf
from documents.tests.utils import DirectoriesMixin
from paperless_tesseract.parsers import RasterisedDocumentParser, get_text_from_pdf, strip_excess_whitespace
image_to_string_calls = []
@ -33,15 +32,32 @@ class FakeImageFile(ContextManager):
return os.path.basename(self.fname)
class TestParser(TestCase):
class TestParser(DirectoriesMixin, TestCase):
def setUp(self):
self.scratch = tempfile.mkdtemp()
text_cases = [
("simple string", "simple string"),
(
"simple newline\n testing string",
"simple newline\ntesting string"
),
(
"utf-8 строка с пробелами в конце ",
"utf-8 строка с пробелами в конце"
)
]
override_settings(SCRATCH_DIR=self.scratch).enable()
def tearDown(self):
shutil.rmtree(self.scratch)
def test_strip_excess_whitespace(self):
for source, result in self.text_cases:
actual_result = strip_excess_whitespace(source)
self.assertEqual(
result,
actual_result,
"strip_exceess_whitespace({}) != '{}', but '{}'".format(
source,
result,
actual_result
)
)
SAMPLE_FILES = os.path.join(os.path.dirname(__file__), "samples")