eynollah/tests/test_deskew.py

13 lines
425 B
Python
Raw Permalink Normal View History

import numpy as np
from eynollah.utils.separate_lines import return_deskew_slop
def test_deskew_landscape_main_page():
# landscape page (width > height) used to crash with
# "TypeError: 'int' object is not iterable" (#223)
img = np.zeros((80, 160))
for y in (20, 40, 60):
img[y:y + 3, 10:150] = 1
angle = return_deskew_slop(img, 2, n_tot_angles=10, main_page=True)
assert np.isfinite(angle)