"""simple docstring"""


import unittest

import numpy as np

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

from ...test_image_processing_common import ImageProcessingSavingTestMixin


if is_torch_available():
    import torch

if is_vision_available():
    from PIL import Image

    from transformers import ChineseCLIPImageProcessor


class 				a (						unittest.TestCase      ):


    def __init__(    self		:     List[str]      , __lowerCAmelCase		:     Any      , __lowerCAmelCase		:     Any=7      , __lowerCAmelCase		:     int=3      , __lowerCAmelCase		:     Optional[int]=18      , __lowerCAmelCase		:     Dict=30      , __lowerCAmelCase		:     Dict=400      , __lowerCAmelCase		:     str=True      , __lowerCAmelCase		:     Tuple=None      , __lowerCAmelCase		:     List[str]=True      , __lowerCAmelCase		:     List[str]=None      , __lowerCAmelCase		:     Tuple=True      , __lowerCAmelCase		:     Union[str, Any]=[0.48_145_466, 0.4_578_275, 0.40_821_073]      , __lowerCAmelCase		:     Tuple=[0.26_862_954, 0.26_130_258, 0.27_577_711]      , __lowerCAmelCase		:     str=True      , ):
      _UpperCAmelCase         =					size if size is not None else {"""height""": 224, """width""": 224}
      _UpperCAmelCase         =					crop_size if crop_size is not None else {"""height""": 18, """width""": 18}
      _UpperCAmelCase         =					parent
      _UpperCAmelCase         =					batch_size
      _UpperCAmelCase         =					num_channels
      _UpperCAmelCase         =					image_size
      _UpperCAmelCase         =					min_resolution
      _UpperCAmelCase         =					max_resolution
      _UpperCAmelCase         =					do_resize
      _UpperCAmelCase         =					size
      _UpperCAmelCase         =					do_center_crop
      _UpperCAmelCase         =					crop_size
      _UpperCAmelCase         =					do_normalize
      _UpperCAmelCase         =					image_mean
      _UpperCAmelCase         =					image_std
      _UpperCAmelCase         =					do_convert_rgb


    def    lowerCAmelCase_     (    self		:     Union[str, Any]    ):
      return {
          "do_resize": self.do_resize,
          "size": self.size,
          "do_center_crop": self.do_center_crop,
          "crop_size": self.crop_size,
          "do_normalize": self.do_normalize,
          "image_mean": self.image_mean,
          "image_std": self.image_std,
          "do_convert_rgb": self.do_convert_rgb,
      }


    def    lowerCAmelCase_     (    self		:     Union[str, Any]      , __lowerCAmelCase		:     List[Any]=False      , __lowerCAmelCase		:     Tuple=False      , __lowerCAmelCase		:     List[Any]=False    ):
      assert not (numpify and torchify), "You cannot specify both numpy and PyTorch tensors at the same time"

      if equal_resolution:
        _UpperCAmelCase         =					[]
        for i in range(self.batch_size    ):
          image_inputs.append(
              np.random.randint(
                  255      , size=(self.num_channels, self.max_resolution, self.max_resolution)      , dtype=np.uinta    )    )
      else:
        _UpperCAmelCase         =					[]
        for i in range(self.batch_size    ):
          _UpperCAmelCase       , _UpperCAmelCase         =					np.random.choice(np.arange(self.min_resolution      , self.max_resolution    )      , 2    )
          image_inputs.append(np.random.randint(255      , size=(self.num_channels, width, height)      , dtype=np.uinta    )    )

      if not numpify and not torchify:
        # PIL expects the channel dimension as last dimension
        _UpperCAmelCase         =					[Image.fromarray(np.moveaxis(__lowerCAmelCase      , 0      , -1    )    ) for x in image_inputs]

      if torchify:
        _UpperCAmelCase         =					[torch.from_numpy(__lowerCAmelCase    ) for x in image_inputs]

      return image_inputs



@require_torch
@require_vision
class 				a (						lowerCAmelCase_		,							unittest.TestCase      ):
    _snake_case   :     Dict   							=							ChineseCLIPImageProcessor if is_vision_available() else None


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      _UpperCAmelCase         =					ChineseCLIPImageProcessingTester(self      , do_center_crop=__lowerCAmelCase    )


    @property
    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      return self.image_processor_tester.prepare_image_processor_dict()


    def    lowerCAmelCase_     (    self		:     int    ):
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_resize"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """size"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_normalize"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """image_mean"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """image_std"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_convert_rgb"""    )    )


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      _UpperCAmelCase         =					self.image_processing_class.from_dict(self.image_processor_dict    )
      self.assertEqual(image_processor.size      , {"""height""": 224, """width""": 224}    )
      self.assertEqual(image_processor.crop_size      , {"""height""": 18, """width""": 18}    )

      _UpperCAmelCase         =					self.image_processing_class.from_dict(self.image_processor_dict      , size=42      , crop_size=84    )
      self.assertEqual(image_processor.size      , {"""shortest_edge""": 42}    )
      self.assertEqual(image_processor.crop_size      , {"""height""": 84, """width""": 84}    )


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      pass


    def    lowerCAmelCase_     (    self		:     Dict    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random PIL images
      _UpperCAmelCase         =					self.image_processor_tester.prepare_inputs(equal_resolution=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , Image.Image    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )


    def    lowerCAmelCase_     (    self		:     str    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random numpy tensors
      _UpperCAmelCase         =					self.image_processor_tester.prepare_inputs(equal_resolution=__lowerCAmelCase      , numpify=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , np.ndarray    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )


    def    lowerCAmelCase_     (    self		:     Union[str, Any]    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random PyTorch tensors
      _UpperCAmelCase         =					self.image_processor_tester.prepare_inputs(equal_resolution=__lowerCAmelCase      , torchify=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , torch.Tensor    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )



@require_torch
@require_vision
class 				a (						lowerCAmelCase_		,							unittest.TestCase      ):
    _snake_case   :     Optional[int]   							=							ChineseCLIPImageProcessor if is_vision_available() else None


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      _UpperCAmelCase         =					ChineseCLIPImageProcessingTester(self      , num_channels=4      , do_center_crop=__lowerCAmelCase    )
      _UpperCAmelCase         =					3


    @property
    def    lowerCAmelCase_     (    self		:     Union[str, Any]    ):
      return self.image_processor_tester.prepare_image_processor_dict()


    def    lowerCAmelCase_     (    self		:     Any    ):
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_resize"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """size"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_normalize"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """image_mean"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """image_std"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_convert_rgb"""    )    )


    def    lowerCAmelCase_     (    self		:     int    ):
      pass


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random PIL images
      _UpperCAmelCase         =					self.image_processor_tester.prepare_inputs(equal_resolution=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , Image.Image    )

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

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , return_tensors="""pt"""    ).pixel_values
      self.assertEqual(
          encoded_images.shape      , (
              self.image_processor_tester.batch_size,
              self.expected_encoded_image_num_channels,
              self.image_processor_tester.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

"""simple docstring"""


import unittest

import numpy as np

from transformers.testing_utils import 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 MobileViTImageProcessor


class 				a (						unittest.TestCase      ):


    def __init__(    self		:     Dict      , __lowerCAmelCase		:     Tuple      , __lowerCAmelCase		:     Optional[Any]=7      , __lowerCAmelCase		:     Optional[Any]=3      , __lowerCAmelCase		:     Optional[Any]=18      , __lowerCAmelCase		:     str=30      , __lowerCAmelCase		:     List[str]=400      , __lowerCAmelCase		:     Union[str, Any]=True      , __lowerCAmelCase		:     str=None      , __lowerCAmelCase		:     Optional[int]=True      , __lowerCAmelCase		:     int=None      , __lowerCAmelCase		:     List[str]=True      , ):
      _UpperCAmelCase         =					size if size is not None else {"""shortest_edge""": 20}
      _UpperCAmelCase         =					crop_size if crop_size is not None else {"""height""": 18, """width""": 18}
      _UpperCAmelCase         =					parent
      _UpperCAmelCase         =					batch_size
      _UpperCAmelCase         =					num_channels
      _UpperCAmelCase         =					image_size
      _UpperCAmelCase         =					min_resolution
      _UpperCAmelCase         =					max_resolution
      _UpperCAmelCase         =					do_resize
      _UpperCAmelCase         =					size
      _UpperCAmelCase         =					do_center_crop
      _UpperCAmelCase         =					crop_size
      _UpperCAmelCase         =					do_flip_channel_order


    def    lowerCAmelCase_     (    self		:     List[str]    ):
      return {
          "do_resize": self.do_resize,
          "size": self.size,
          "do_center_crop": self.do_center_crop,
          "crop_size": self.crop_size,
          "do_flip_channel_order": self.do_flip_channel_order,
      }



@require_torch
@require_vision
class 				a (						lowerCAmelCase_		,							unittest.TestCase      ):
    _snake_case   :     Optional[int]   							=							MobileViTImageProcessor if is_vision_available() else None


    def    lowerCAmelCase_     (    self		:     Optional[Any]    ):
      _UpperCAmelCase         =					MobileViTImageProcessingTester(self    )


    @property
    def    lowerCAmelCase_     (    self		:     Tuple    ):
      return self.image_processor_tester.prepare_image_processor_dict()


    def    lowerCAmelCase_     (    self		:     Dict    ):
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_resize"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """size"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """center_crop"""    )    )
      self.assertTrue(hasattr(__lowerCAmelCase      , """do_flip_channel_order"""    )    )


    def    lowerCAmelCase_     (    self		:     Union[str, Any]    ):
      _UpperCAmelCase         =					self.image_processing_class.from_dict(self.image_processor_dict    )
      self.assertEqual(image_processor.size      , {"""shortest_edge""": 20}    )
      self.assertEqual(image_processor.crop_size      , {"""height""": 18, """width""": 18}    )

      _UpperCAmelCase         =					self.image_processing_class.from_dict(self.image_processor_dict      , size=42      , crop_size=84    )
      self.assertEqual(image_processor.size      , {"""shortest_edge""": 42}    )
      self.assertEqual(image_processor.crop_size      , {"""height""": 84, """width""": 84}    )


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


    def    lowerCAmelCase_     (    self		:     Dict    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random PIL images
      _UpperCAmelCase         =					prepare_image_inputs(self.image_processor_tester      , equal_resolution=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , Image.Image    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )


    def    lowerCAmelCase_     (    self		:     str    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random numpy tensors
      _UpperCAmelCase         =					prepare_image_inputs(self.image_processor_tester      , equal_resolution=__lowerCAmelCase      , numpify=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , np.ndarray    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )


    def    lowerCAmelCase_     (    self		:     Optional[int]    ):
      # Initialize image_processing
      _UpperCAmelCase         =					self.image_processing_class(**self.image_processor_dict    )
      # create random PyTorch tensors
      _UpperCAmelCase         =					prepare_image_inputs(self.image_processor_tester      , equal_resolution=__lowerCAmelCase      , torchify=__lowerCAmelCase    )
      for image in image_inputs:
        self.assertIsInstance(__lowerCAmelCase      , torch.Tensor    )

      # Test not batched input
      _UpperCAmelCase         =					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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )

      # Test batched
      _UpperCAmelCase         =					image_processing(__lowerCAmelCase      , 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.crop_size["""height"""],
              self.image_processor_tester.crop_size["""width"""],
          )      , )
