"""simple docstring"""
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    ,			lowercase    ,			lowercase=7    ,			lowercase=3    ,			lowercase=18    ,			lowercase=30    ,			lowercase=4_00    ,			lowercase=True    ,			lowercase=None    ,			lowercase=True    ,			lowercase=False    ,			lowercase=True    ,			lowercase=True    ,			lowercase=[0.5, 0.5, 0.5]    ,			lowercase=[0.5, 0.5, 0.5]    ,			)	->			Tuple:





        '''simple docstring'''





        a__:     Any										=	parent
        a__:     List[Any]										=	batch_size
        a__:     Dict										=	num_channels
        a__:     Optional[Any]										=	image_size
        a__:     Optional[int]										=	min_resolution
        a__:     Union[str, Any]										=	max_resolution
        a__:     Any										=	do_resize
        a__:     Tuple										=	size if size is not None else {'height': 18, 'width': 20}
        a__:     int										=	do_thumbnail
        a__:     Union[str, Any]										=	do_align_axis
        a__:     Optional[Any]										=	do_pad
        a__:     Dict										=	do_normalize
        a__:     str										=	image_mean
        a__:     Union[str, Any]										=	image_std



      def   lowerCamelCase_   (						self)	->			Optional[Any]:





        '''simple docstring'''





        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       (     __lowerCAmelCase       ,      unittest.TestCase				):
      a__        =		DonutImageProcessor if is_vision_available() else None



      def   lowerCamelCase_   (						self)	->			str:





        '''simple docstring'''





        a__:     str										=	DonutImageProcessingTester(self)



      @property
      def   lowerCamelCase_   (						self)	->			Optional[int]:





        '''simple docstring'''





        return self.image_processor_tester.prepare_image_processor_dict()



      def   lowerCamelCase_   (						self)	->			Optional[Any]:





        '''simple docstring'''





        a__:     str										=	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   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     List[str]										=	self.image_processing_class.from_dict(self.image_processor_dict)
        self.assertEqual(image_processor.size    ,			{'height': 18, 'width': 20})

        a__:     Optional[int]										=	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
        a__:     Optional[int]										=	self.image_processing_class.from_dict(self.image_processor_dict    ,			size=(42, 84))
        self.assertEqual(image_processor.size    ,			{'height': 84, 'width': 42})



      def   lowerCamelCase_   (						self)	->			List[str]:





        '''simple docstring'''





        pass



      @is_flaky()
      def   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     Dict										=	self.image_processing_class(**self.image_processor_dict)
        # create random PIL images
        a__:     Optional[int]										=	prepare_image_inputs(self.image_processor_tester    ,			equal_resolution=lowercase)
        for image in image_inputs:
          self.assertIsInstance(lowercase    ,			Image.Image)

        # Test not batched input
        a__:     int										=	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
        a__:     Dict										=	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   lowerCamelCase_   (						self)	->			Optional[int]:





        '''simple docstring'''





        a__:     str										=	self.image_processing_class(**self.image_processor_dict)
        # create random numpy tensors
        a__:     str										=	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
        a__:     str										=	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
        a__:     Tuple										=	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   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     Optional[Any]										=	self.image_processing_class(**self.image_processor_dict)
        # create random PyTorch tensors
        a__:     str										=	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
        a__:     Optional[int]										=	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
        a__:     List[Any]										=	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'],
            )    ,			)



"""simple docstring"""
import shutil
import tempfile
import unittest

import numpy as np

from transformers.testing_utils import (
    is_pt_tf_cross_test,
    require_tf,
    require_torch,
    require_torchvision,
    require_vision,
)
from transformers.utils import is_tf_available, is_torch_available, is_vision_available


if is_vision_available():
 from PIL import Image

 from transformers import AutoProcessor, SamImageProcessor, SamProcessor

if is_torch_available():
 import torch

if is_tf_available():
 import tensorflow as tf




@require_vision
@require_torchvision
class    __snake_case       (     unittest.TestCase				):



      def   lowerCamelCase_   (						self)	->			Any:





        '''simple docstring'''





        a__:     Optional[Any]										=	tempfile.mkdtemp()
        a__:     Optional[int]										=	SamImageProcessor()
        a__:     Tuple										=	SamProcessor(lowercase)
        processor.save_pretrained(self.tmpdirname)



      def   lowerCamelCase_   (						self    ,			**lowercase)	->			List[Any]:





        '''simple docstring'''





        return AutoProcessor.from_pretrained(self.tmpdirname    ,			**lowercase).image_processor



      def   lowerCamelCase_   (						self)	->			Optional[int]:





        '''simple docstring'''





        shutil.rmtree(self.tmpdirname)



      def   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     Any										=	[np.random.randint(2_55    ,			size=(3, 30, 4_00)    ,			dtype=np.uinta)]

        a__:     Optional[Any]										=	[Image.fromarray(np.moveaxis(lowercase    ,			0    ,			-1)) for x in image_inputs]

        return image_inputs



      def   lowerCamelCase_   (						self)	->			str:





        '''simple docstring'''





        a__:     List[str]										=	SamProcessor(image_processor=self.get_image_processor())
        processor.save_pretrained(self.tmpdirname)

        a__:     Optional[int]										=	self.get_image_processor(do_normalize=lowercase    ,			padding_value=1.0)

        a__:     List[Any]										=	SamProcessor.from_pretrained(self.tmpdirname    ,			do_normalize=lowercase    ,			padding_value=1.0)

        self.assertEqual(processor.image_processor.to_json_string()    ,			image_processor_add_kwargs.to_json_string())
        self.assertIsInstance(processor.image_processor    ,			lowercase)



      def   lowerCamelCase_   (						self)	->			Tuple:





        '''simple docstring'''





        a__:     Union[str, Any]										=	self.get_image_processor()

        a__:     List[Any]										=	SamProcessor(image_processor=lowercase)

        a__:     Optional[int]										=	self.prepare_image_inputs()

        a__:     Optional[Any]										=	image_processor(lowercase    ,			return_tensors='np')
        a__:     Tuple										=	processor(images=lowercase    ,			return_tensors='np')

        input_feat_extract.pop('original_sizes')  # pop original_sizes as it is popped in the processor
        input_feat_extract.pop('reshaped_input_sizes')  # pop original_sizes as it is popped in the processor

        for key in input_feat_extract.keys():
          self.assertAlmostEqual(input_feat_extract[key].sum()    ,			input_processor[key].sum()    ,			delta=1e-2)



      @require_torch
      def   lowerCamelCase_   (						self)	->			Any:





        '''simple docstring'''





        a__:     int										=	self.get_image_processor()

        a__:     List[str]										=	SamProcessor(image_processor=lowercase)
        a__:     Optional[Any]										=	[torch.ones((1, 3, 5, 5))]

        a__:     Union[str, Any]										=	[[17_64, 26_46]]

        a__:     Optional[Any]										=	[[6_83, 10_24]]
        a__:     int										=	processor.post_process_masks(lowercase    ,			lowercase    ,			lowercase)
        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        a__:     Optional[int]										=	processor.post_process_masks(
            lowercase    ,			torch.tensor(lowercase)    ,			torch.tensor(lowercase))
        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        # should also work with np
        a__:     Dict										=	[np.ones((1, 3, 5, 5))]
        a__:     Tuple										=	processor.post_process_masks(lowercase    ,			np.array(lowercase)    ,			np.array(lowercase))

        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        a__:     Tuple										=	[[1, 0], [0, 1]]
        with self.assertRaises(lowercase):
          a__:     List[Any]										=	processor.post_process_masks(lowercase    ,			np.array(lowercase)    ,			np.array(lowercase))







@require_vision
@require_tf
class    __snake_case       (     unittest.TestCase				):



      def   lowerCamelCase_   (						self)	->			List[Any]:





        '''simple docstring'''





        a__:     Optional[Any]										=	tempfile.mkdtemp()
        a__:     List[Any]										=	SamImageProcessor()
        a__:     Optional[int]										=	SamProcessor(lowercase)
        processor.save_pretrained(self.tmpdirname)



      def   lowerCamelCase_   (						self    ,			**lowercase)	->			int:





        '''simple docstring'''





        return AutoProcessor.from_pretrained(self.tmpdirname    ,			**lowercase).image_processor



      def   lowerCamelCase_   (						self)	->			Tuple:





        '''simple docstring'''





        shutil.rmtree(self.tmpdirname)



      def   lowerCamelCase_   (						self)	->			Tuple:





        '''simple docstring'''





        a__:     Optional[Any]										=	[np.random.randint(2_55    ,			size=(3, 30, 4_00)    ,			dtype=np.uinta)]

        a__:     Optional[int]										=	[Image.fromarray(np.moveaxis(lowercase    ,			0    ,			-1)) for x in image_inputs]

        return image_inputs



      def   lowerCamelCase_   (						self)	->			Any:





        '''simple docstring'''





        a__:     List[str]										=	SamProcessor(image_processor=self.get_image_processor())
        processor.save_pretrained(self.tmpdirname)

        a__:     Dict										=	self.get_image_processor(do_normalize=lowercase    ,			padding_value=1.0)

        a__:     Union[str, Any]										=	SamProcessor.from_pretrained(self.tmpdirname    ,			do_normalize=lowercase    ,			padding_value=1.0)

        self.assertEqual(processor.image_processor.to_json_string()    ,			image_processor_add_kwargs.to_json_string())
        self.assertIsInstance(processor.image_processor    ,			lowercase)



      def   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     Optional[Any]										=	self.get_image_processor()

        a__:     str										=	SamProcessor(image_processor=lowercase)

        a__:     int										=	self.prepare_image_inputs()

        a__:     int										=	image_processor(lowercase    ,			return_tensors='np')
        a__:     Dict										=	processor(images=lowercase    ,			return_tensors='np')

        input_feat_extract.pop('original_sizes')  # pop original_sizes as it is popped in the processor
        input_feat_extract.pop('reshaped_input_sizes')  # pop reshaped_input_sizes as it is popped in the processor

        for key in input_feat_extract.keys():
          self.assertAlmostEqual(input_feat_extract[key].sum()    ,			input_processor[key].sum()    ,			delta=1e-2)



      @require_tf
      def   lowerCamelCase_   (						self)	->			Any:





        '''simple docstring'''





        a__:     Tuple										=	self.get_image_processor()

        a__:     Any										=	SamProcessor(image_processor=lowercase)
        a__:     str										=	[tf.ones((1, 3, 5, 5))]

        a__:     List[Any]										=	[[17_64, 26_46]]

        a__:     List[Any]										=	[[6_83, 10_24]]
        a__:     List[Any]										=	processor.post_process_masks(lowercase    ,			lowercase    ,			lowercase    ,			return_tensors='tf')
        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        a__:     Tuple										=	processor.post_process_masks(
            lowercase    ,			tf.convert_to_tensor(lowercase)    ,			tf.convert_to_tensor(lowercase)    ,			return_tensors='tf'    ,			)
        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        # should also work with np
        a__:     Optional[Any]										=	[np.ones((1, 3, 5, 5))]
        a__:     int										=	processor.post_process_masks(
            lowercase    ,			np.array(lowercase)    ,			np.array(lowercase)    ,			return_tensors='tf')

        self.assertEqual(masks[0].shape    ,			(1, 3, 17_64, 26_46))

        a__:     List[str]										=	[[1, 0], [0, 1]]
        with self.assertRaises(tf.errors.InvalidArgumentError):
          a__:     Any										=	processor.post_process_masks(
              lowercase    ,			np.array(lowercase)    ,			np.array(lowercase)    ,			return_tensors='tf')







@require_vision
@require_torchvision
class    __snake_case       (     unittest.TestCase				):



      def   lowerCamelCase_   (						self)	->			Dict:





        '''simple docstring'''





        a__:     str										=	tempfile.mkdtemp()
        a__:     int										=	SamImageProcessor()
        a__:     Union[str, Any]										=	SamProcessor(lowercase)
        processor.save_pretrained(self.tmpdirname)



      def   lowerCamelCase_   (						self    ,			**lowercase)	->			Optional[int]:





        '''simple docstring'''





        return AutoProcessor.from_pretrained(self.tmpdirname    ,			**lowercase).image_processor



      def   lowerCamelCase_   (						self)	->			Tuple:





        '''simple docstring'''





        shutil.rmtree(self.tmpdirname)



      def   lowerCamelCase_   (						self)	->			Tuple:





        '''simple docstring'''





        a__:     Any										=	[np.random.randint(2_55    ,			size=(3, 30, 4_00)    ,			dtype=np.uinta)]

        a__:     Any										=	[Image.fromarray(np.moveaxis(lowercase    ,			0    ,			-1)) for x in image_inputs]

        return image_inputs



      @is_pt_tf_cross_test
      def   lowerCamelCase_   (						self)	->			int:





        '''simple docstring'''





        a__:     Optional[int]										=	self.get_image_processor()

        a__:     int										=	SamProcessor(image_processor=lowercase)
        a__:     int										=	np.random.randint(0    ,			2    ,			size=(1, 3, 5, 5)).astype(np.floataa)
        a__:     Dict										=	[tf.convert_to_tensor(lowercase)]
        a__:     Union[str, Any]										=	[torch.tensor(lowercase)]

        a__:     List[Any]										=	[[17_64, 26_46]]

        a__:     Optional[Any]										=	[[6_83, 10_24]]
        a__:     Tuple										=	processor.post_process_masks(
            lowercase    ,			lowercase    ,			lowercase    ,			return_tensors='tf')
        a__:     str										=	processor.post_process_masks(
            lowercase    ,			lowercase    ,			lowercase    ,			return_tensors='pt')

        self.assertTrue(np.all(tf_masks[0].numpy() == pt_masks[0].numpy()))



      @is_pt_tf_cross_test
      def   lowerCamelCase_   (						self)	->			str:





        '''simple docstring'''





        a__:     Tuple										=	self.get_image_processor()

        a__:     Dict										=	SamProcessor(image_processor=lowercase)

        a__:     Any										=	self.prepare_image_inputs()

        a__:     List[Any]										=	image_processor(lowercase    ,			return_tensors='pt')['pixel_values'].numpy()
        a__:     Tuple										=	processor(images=lowercase    ,			return_tensors='pt')['pixel_values'].numpy()

        a__:     Any										=	image_processor(lowercase    ,			return_tensors='tf')['pixel_values'].numpy()
        a__:     Any										=	processor(images=lowercase    ,			return_tensors='tf')['pixel_values'].numpy()

        self.assertTrue(np.allclose(lowercase    ,			lowercase))
        self.assertTrue(np.allclose(lowercase    ,			lowercase))
        self.assertTrue(np.allclose(lowercase    ,			lowercase))


