mirror of https://github.com/open-mmlab/mmocr.git
19 lines
494 B
Python
19 lines
494 B
Python
"""Test orientation check and ignore method."""
|
|
|
|
import shutil
|
|
import tempfile
|
|
|
|
from mmocr.utils import drop_orientation
|
|
|
|
|
|
def test_drop_orientation():
|
|
img_file = 'tests/data/test_img2.jpg'
|
|
output_file = drop_orientation(img_file)
|
|
assert output_file is img_file
|
|
|
|
img_file = 'tests/data/test_img1.jpg'
|
|
tmp_dir = tempfile.TemporaryDirectory()
|
|
dst_file = shutil.copy(img_file, tmp_dir.name)
|
|
output_file = drop_orientation(dst_file)
|
|
assert output_file[-3:] == 'png'
|