Add another test

This commit is contained in:
Jim Barlow
2015-07-22 03:16:19 -07:00
parent ec8a35a7a6
commit ce2dbdf372
4 changed files with 23 additions and 7 deletions
+2 -1
View File
@@ -1,3 +1,4 @@
tmp/
log/
*.pyc
*.pyc
tests/output/
+8 -1
View File
@@ -1 +1,8 @@
All test resources must come from public domain sources such as Wikimedia.
All test resources must come from free public domain sources for copyright reasons.
Test files do not necessarily produce perfect (or even good) OCR results.
File | Source
--------------------|---------
graph.pdf | Wikimedia
c02-22.pdf | Project Gutenberg: https://www.gutenberg.org/files/76/76-h/images/c02-22.jpg
Binary file not shown.
+13 -5
View File
@@ -7,11 +7,19 @@ TESTS_ROOT = os.path.abspath(os.path.dirname(__file__))
PROJECT_ROOT = os.path.dirname(TESTS_ROOT)
OCRMYPDF = os.path.join(PROJECT_ROOT, 'OCRmyPDF.sh')
TEST_RESOURCES = os.path.join(PROJECT_ROOT, 'tests', 'resources')
TEST_OUTPUT = os.path.join(PROJECT_ROOT, 'tests', 'output')
def run_ocrmypdf(input_file, output_file, *args):
def setup_module():
try:
os.mkdir(TEST_OUTPUT)
except FileExistsError:
pass
def run_ocrmypdf(input_file, *args):
input_path = os.path.join(TEST_RESOURCES, input_file)
output_path = os.path.join(TEST_RESOURCES, output_file)
output_path = os.path.join(TEST_OUTPUT, input_file)
sh_args = ['sh', './OCRmyPDF.sh'] + list(args) + [input_path, output_path]
sh = Popen(
@@ -21,10 +29,10 @@ def run_ocrmypdf(input_file, output_file, *args):
return sh, out, err
def check_ocrmypdf(input_file, output_file, *args):
sh, _, err = run_ocrmypdf(input_file, output_file, *args)
def check_ocrmypdf(input_file, *args):
sh, _, err = run_ocrmypdf(input_file, *args)
assert sh.returncode == 0, err
def test_quick():
check_ocrmypdf('graph.pdf', 'graph_out.pdf')
check_ocrmypdf('c02-22.pdf')