



from ..utils import DummyObject, requires_backends
class 				_snake_case			(					metaclass=lowercase__):
				UpperCamelCase__				:					Union[str, Any]     		=["""speech"""]




				def __init__(						self					:						List[Any],					*__lowercase					:						Optional[Any],					**__lowercase					:						str						):
							requires_backends(self,					["speech"]						)
class 				_snake_case			(					metaclass=lowercase__):
				UpperCamelCase__				:					Optional[Any]     		=["""speech"""]




				def __init__(						self					:						Union[str, Any],					*__lowercase					:						List[Any],					**__lowercase					:						Tuple						):
							requires_backends(self,					["speech"]						)









import unittest

import numpy as np

from transformers.testing_utils import is_flaky, require_torch, require_vision
from transformers.utils import is_torch_available, is_vision_available

from ...test_image_processing_common import ImageProcessingSavingTestMixin, prepare_image_inputs


if is_torch_available():
							import torch

if is_vision_available():
							from PIL import Image

							from transformers import DonutImageProcessor
class 				_snake_case			(					unittest.TestCase):




				def __init__(						self					:						Dict,					__lowercase					:						int,					__lowercase					:						Union[str, Any]=7,					__lowercase					:						Union[str, Any]=3,					__lowercase					:						Any=18,					__lowercase					:						Union[str, Any]=30,					__lowercase					:						Any=400,					__lowercase					:						List[str]=True,					__lowercase					:						Dict=None,					__lowercase					:						List[str]=True,					__lowercase					:						int=False,					__lowercase					:						Union[str, Any]=True,					__lowercase					:						str=True,					__lowercase					:						Optional[int]=[0.5, 0.5, 0.5],					__lowercase					:						List[Any]=[0.5, 0.5, 0.5],					):
							lowercase__					  =				parent
							lowercase__					  =				batch_size
							lowercase__					  =				num_channels
							lowercase__					  =				image_size
							lowercase__					  =				min_resolution
							lowercase__					  =				max_resolution
							lowercase__					  =				do_resize
							lowercase__					  =				size if size is not None else {"height": 18, "width": 20}
							lowercase__					  =				do_thumbnail
							lowercase__					  =				do_align_axis
							lowercase__					  =				do_pad
							lowercase__					  =				do_normalize
							lowercase__					  =				image_mean
							lowercase__					  =				image_std




				def 							A__      (						self					:						Optional[Any]						):
							return {
							    "do_resize": self.do_resize,
							    "size": self.size,
							    "do_thumbnail": self.do_thumbnail,
							    "do_align_long_axis": self.do_align_axis,
							    "do_pad": self.do_pad,
							    "do_normalize": self.do_normalize,
							    "image_mean": self.image_mean,
							    "image_std": self.image_std,
							}
@require_torch
@require_vision
class 				_snake_case			(					lowercase__    ,	unittest.TestCase):
				UpperCamelCase__				:					Optional[int]     		=DonutImageProcessor if is_vision_available() else None




				def 							A__      (						self					:						str						):
							lowercase__					  =				DonutImageProcessingTester(self						)




				@property
				def 							A__      (						self					:						List[str]						):
							return self.image_processor_tester.prepare_image_processor_dict()




				def 							A__      (						self					:						Optional[Any]						):
							lowercase__					  =				self.image_processing_class(**self.image_processor_dict						)
							self.assertTrue(hasattr(__lowercase,					"do_resize"						)						)
							self.assertTrue(hasattr(__lowercase,					"size"						)						)
							self.assertTrue(hasattr(__lowercase,					"do_thumbnail"						)						)
							self.assertTrue(hasattr(__lowercase,					"do_align_long_axis"						)						)
							self.assertTrue(hasattr(__lowercase,					"do_pad"						)						)
							self.assertTrue(hasattr(__lowercase,					"do_normalize"						)						)
							self.assertTrue(hasattr(__lowercase,					"image_mean"						)						)
							self.assertTrue(hasattr(__lowercase,					"image_std"						)						)




				def 							A__      (						self					:						str						):
							lowercase__					  =				self.image_processing_class.from_dict(self.image_processor_dict						)
							self.assertEqual(image_processor.size,					{"height": 18, "width": 20}						)

							lowercase__					  =				self.image_processing_class.from_dict(self.image_processor_dict,					size=42						)
							self.assertEqual(image_processor.size,					{"height": 42, "width": 42}						)

							# Previous config had dimensions in (width, height) order
							lowercase__					  =				self.image_processing_class.from_dict(self.image_processor_dict,					size=(42, 84)						)
							self.assertEqual(image_processor.size,					{"height": 84, "width": 42}						)




				def 							A__      (						self					:						List[str]						):
							pass




				@is_flaky()
				def 							A__      (						self					:						Dict						):
							# Initialize image_processing
							lowercase__					  =				self.image_processing_class(**self.image_processor_dict						)
							# create random PIL images
							lowercase__					  =				prepare_image_inputs(self.image_processor_tester,					equal_resolution=__lowercase						)
							for image in image_inputs:
										self.assertIsInstance(__lowercase,					Image.Image						)

							# Test not batched input
							lowercase__					  =				image_processing(image_inputs[0],					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)

							# Test batched
							lowercase__					  =				image_processing(__lowercase,					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)




				@is_flaky()
				def 							A__      (						self					:						Optional[Any]						):
							# Initialize image_processing
							lowercase__					  =				self.image_processing_class(**self.image_processor_dict						)
							# create random numpy tensors
							lowercase__					  =				prepare_image_inputs(self.image_processor_tester,					equal_resolution=__lowercase,					numpify=__lowercase						)
							for image in image_inputs:
										self.assertIsInstance(__lowercase,					np.ndarray						)

							# Test not batched input
							lowercase__					  =				image_processing(image_inputs[0],					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)

							# Test batched
							lowercase__					  =				image_processing(__lowercase,					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)




				@is_flaky()
				def 							A__      (						self					:						Tuple						):
							# Initialize image_processing
							lowercase__					  =				self.image_processing_class(**self.image_processor_dict						)
							# create random PyTorch tensors
							lowercase__					  =				prepare_image_inputs(self.image_processor_tester,					equal_resolution=__lowercase,					torchify=__lowercase						)
							for image in image_inputs:
										self.assertIsInstance(__lowercase,					torch.Tensor						)

							# Test not batched input
							lowercase__					  =				image_processing(image_inputs[0],					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)

							# Test batched
							lowercase__					  =				image_processing(__lowercase,					return_tensors="pt"						).pixel_values
							self.assertEqual(
							    encoded_images.shape,					(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size["height"],
							        self.image_processor_tester.size["width"],
							    ),					)




