



from collections.abc import Generator
from math import sin





def   __snake_case			(    _UpperCAmelCase       ):
    if len(_UpperCAmelCase       ) != 32:
        raise ValueError('''Input must be of length 32'''       )

    __a		 =       b''''''
    for i in [3, 2, 1, 0]:
        little_endian += string_aa[8 * i : 8 * i + 8]
    return little_endian





def   __snake_case			(    _UpperCAmelCase       ):
    if i < 0:
        raise ValueError('''Input must be non-negative'''       )

    __a		 =       format(_UpperCAmelCase     ,				'''08x'''       )[-8:]
    __a		 =       b''''''
    for i in [3, 2, 1, 0]:
        little_endian_hex += hex_rep[2 * i : 2 * i + 2].encode('''utf-8'''       )
    return little_endian_hex





def   __snake_case			(    _UpperCAmelCase       ):
    __a		 =       b''''''
    for char in message:
        bit_string += format(_UpperCAmelCase     ,				'''08b'''       ).encode('''utf-8'''       )
    __a		 =       format(len(_UpperCAmelCase       )     ,				'''064b'''       ).encode('''utf-8'''       )

    # Pad bit_string to a multiple of 512 chars
    bit_string += b"1"
    while len(_UpperCAmelCase       ) % 512 != 448:
        bit_string += b"0"
    bit_string += to_little_endian(start_len[32:]       ) + to_little_endian(start_len[:32]       )

    return bit_string





def   __snake_case			(    _UpperCAmelCase       ):
    if len(_UpperCAmelCase       ) % 512 != 0:
        raise ValueError('''Input must have length that\'s a multiple of 512'''       )

    for pos in range(0     ,				len(_UpperCAmelCase       )     ,				512       ):
        __a		 =       bit_string[pos : pos + 512]
        __a		 =       []
        for i in range(0     ,				512     ,				32       ):
            block_words.append(int(to_little_endian(block[i : i + 32]       )     ,				2       )       )
        yield block_words





def   __snake_case			(    _UpperCAmelCase       ):
    if i < 0:
        raise ValueError('''Input must be non-negative'''       )

    __a		 =       format(_UpperCAmelCase     ,				'''032b'''       )
    __a		 =       ''''''
    for c in i_str:
        new_str += "1" if c == "0" else "0"
    return int(_UpperCAmelCase     ,				2       )





def   __snake_case			(    _UpperCAmelCase     ,				_UpperCAmelCase       ):
    return (a + b) % 2**32





def   __snake_case			(    _UpperCAmelCase     ,				_UpperCAmelCase       ):
    if i < 0:
        raise ValueError('''Input must be non-negative'''       )
    if shift < 0:
        raise ValueError('''Shift must be non-negative'''       )
    return ((i << shift) ^ (i >> (32 - shift))) % 2**32





def   __snake_case			(    _UpperCAmelCase       ):
    __a		 =       preprocess(_UpperCAmelCase       )

    __a		 =       [int(2**32 * abs(sin(i + 1       )       )       ) for i in range(64       )]

    # Starting states
    __a		 =       0X67_452_301
    __a		 =       0Xef_cda_b89
    __a		 =       0X98_bad_cfe
    __a		 =       0X10_325_476

    __a		 =       [
        7,
        12,
        17,
        22,
        7,
        12,
        17,
        22,
        7,
        12,
        17,
        22,
        7,
        12,
        17,
        22,
        5,
        9,
        14,
        20,
        5,
        9,
        14,
        20,
        5,
        9,
        14,
        20,
        5,
        9,
        14,
        20,
        4,
        11,
        16,
        23,
        4,
        11,
        16,
        23,
        4,
        11,
        16,
        23,
        4,
        11,
        16,
        23,
        6,
        10,
        15,
        21,
        6,
        10,
        15,
        21,
        6,
        10,
        15,
        21,
        6,
        10,
        15,
        21,
    ]

    # Process bit string in chunks, each with 16 32-char words
    for block_words in get_block_words(_UpperCAmelCase       ):
        __a		 =       aa
        __a		 =       ba
        __a		 =       ca
        __a		 =       da

        # Hash current chunk
        for i in range(64       ):
            if i <= 15:
                # f = (b & c) | (not_32(b) & d)     # Alternate definition for f
                __a		 =       d ^ (b & (c ^ d))
                __a		 =       i
            elif i <= 31:
                # f = (d & b) | (not_32(d) & c)     # Alternate definition for f
                __a		 =       c ^ (d & (b ^ c))
                __a		 =       (5 * i + 1) % 16
            elif i <= 47:
                __a		 =       b ^ c ^ d
                __a		 =       (3 * i + 5) % 16
            else:
                __a		 =       c ^ (b | not_aa(_UpperCAmelCase       ))
                __a		 =       (7 * i) % 16
            __a		 =       (f + a + added_consts[i] + block_words[g]) % 2**32
            __a		 =       d
            __a		 =       c
            __a		 =       b
            __a		 =       sum_aa(_UpperCAmelCase     ,				left_rotate_aa(_UpperCAmelCase     ,				shift_amounts[i]       )       )

        # Add hashed chunk to running total
        __a		 =       sum_aa(_UpperCAmelCase     ,				_UpperCAmelCase       )
        __a		 =       sum_aa(_UpperCAmelCase     ,				_UpperCAmelCase       )
        __a		 =       sum_aa(_UpperCAmelCase     ,				_UpperCAmelCase       )
        __a		 =       sum_aa(_UpperCAmelCase     ,				_UpperCAmelCase       )

    __a		 =       reformat_hex(_UpperCAmelCase       ) + reformat_hex(_UpperCAmelCase       ) + reformat_hex(_UpperCAmelCase       ) + reformat_hex(_UpperCAmelCase       )
    return digest


if __name__ == "__main__":
 import doctest

 doctest.testmod()










import unittest

import numpy as np
from datasets import load_dataset

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 BeitImageProcessor

class 					_A (			unittest.TestCase					):



       def __init__(	self							:  Union[str, Any]  ,	__SCREAMING_SNAKE_CASE							:  List[str]  ,	__SCREAMING_SNAKE_CASE							:  int=7  ,	__SCREAMING_SNAKE_CASE							:  Tuple=3  ,	__SCREAMING_SNAKE_CASE							:  List[Any]=18  ,	__SCREAMING_SNAKE_CASE							:  Optional[Any]=30  ,	__SCREAMING_SNAKE_CASE							:  int=400  ,	__SCREAMING_SNAKE_CASE							:  Dict=True  ,	__SCREAMING_SNAKE_CASE							:  str=None  ,	__SCREAMING_SNAKE_CASE							:  Union[str, Any]=True  ,	__SCREAMING_SNAKE_CASE							:  Union[str, Any]=None  ,	__SCREAMING_SNAKE_CASE							:  str=True  ,	__SCREAMING_SNAKE_CASE							:  Union[str, Any]=[0.5, 0.5, 0.5]  ,	__SCREAMING_SNAKE_CASE							:  Any=[0.5, 0.5, 0.5]  ,	__SCREAMING_SNAKE_CASE							:  List[str]=False  ,	):


           '''simple docstring'''

           __a		 =       size if size is not None else {'''height''': 20, '''width''': 20}
           __a		 =       crop_size if crop_size is not None else {'''height''': 18, '''width''': 18}
           __a		 =       parent
           __a		 =       batch_size
           __a		 =       num_channels
           __a		 =       image_size
           __a		 =       min_resolution
           __a		 =       max_resolution
           __a		 =       do_resize
           __a		 =       size
           __a		 =       do_center_crop
           __a		 =       crop_size
           __a		 =       do_normalize
           __a		 =       image_mean
           __a		 =       image_std
           __a		 =       do_reduce_labels





       def 	_lowerCamelCase    (	self							:  str):


           '''simple docstring'''

           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_reduce_labels": self.do_reduce_labels,
           }





def   __snake_case			(    ):
    __a		 =       load_dataset('''hf-internal-testing/fixtures_ade20k'''     ,				split='''test'''       )

    __a		 =       Image.open(dataset[0]['''file''']       )
    __a		 =       Image.open(dataset[1]['''file''']       )

    return image, map





def   __snake_case			(    ):
    __a		 =       load_dataset('''hf-internal-testing/fixtures_ade20k'''     ,				split='''test'''       )

    __a		 =       Image.open(ds[0]['''file''']       )
    __a		 =       Image.open(ds[1]['''file''']       )
    __a		 =       Image.open(ds[2]['''file''']       )
    __a		 =       Image.open(ds[3]['''file''']       )

    return [imagea, imagea], [mapa, mapa]

@require_torch
@require_vision
class 					_A (			__UpperCAmelCase   ,unittest.TestCase					):
       UpperCamelCase__						:					Union[str, Any]		      =		BeitImageProcessor if is_vision_available() else None



       def 	_lowerCamelCase    (	self							:  int):


           '''simple docstring'''

           __a		 =       BeitImageProcessingTester(self)



       @property
       def 	_lowerCamelCase    (	self							:  int):


           '''simple docstring'''

           return self.image_processor_tester.prepare_image_processor_dict()



       def 	_lowerCamelCase    (	self							:  Optional[Any]):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''do_resize'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''size'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''do_center_crop'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''center_crop'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''do_normalize'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''image_mean'''))
           self.assertTrue(hasattr(__SCREAMING_SNAKE_CASE  ,	'''image_std'''))



       def 	_lowerCamelCase    (	self							:  str):


           '''simple docstring'''

           __a		 =       self.image_processing_class.from_dict(self.image_processor_dict)
           self.assertEqual(image_processor.size  ,	{'''height''': 20, '''width''': 20})
           self.assertEqual(image_processor.crop_size  ,	{'''height''': 18, '''width''': 18})
           self.assertEqual(image_processor.do_reduce_labels  ,	__SCREAMING_SNAKE_CASE)

           __a		 =       self.image_processing_class.from_dict(
               self.image_processor_dict  ,	size=42  ,	crop_size=84  ,	reduce_labels=__SCREAMING_SNAKE_CASE)
           self.assertEqual(image_processor.size  ,	{'''height''': 42, '''width''': 42})
           self.assertEqual(image_processor.crop_size  ,	{'''height''': 84, '''width''': 84})
           self.assertEqual(image_processor.do_reduce_labels  ,	__SCREAMING_SNAKE_CASE)



       def 	_lowerCamelCase    (	self							:  Dict):


           '''simple docstring'''

           pass



       def 	_lowerCamelCase    (	self							:  Tuple):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)
           # create random PIL images
           __a		 =       prepare_image_inputs(self.image_processor_tester  ,	equal_resolution=__SCREAMING_SNAKE_CASE)
           for image in image_inputs:
               self.assertIsInstance(__SCREAMING_SNAKE_CASE  ,	Image.Image)

           # Test not batched input
           __a		 =       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
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	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							:  int):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)
           # create random numpy tensors
           __a		 =       prepare_image_inputs(self.image_processor_tester  ,	equal_resolution=__SCREAMING_SNAKE_CASE  ,	numpify=__SCREAMING_SNAKE_CASE)
           for image in image_inputs:
               self.assertIsInstance(__SCREAMING_SNAKE_CASE  ,	np.ndarray)

           # Test not batched input
           __a		 =       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
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	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							:  List[Any]):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)
           # create random PyTorch tensors
           __a		 =       prepare_image_inputs(self.image_processor_tester  ,	equal_resolution=__SCREAMING_SNAKE_CASE  ,	torchify=__SCREAMING_SNAKE_CASE)
           for image in image_inputs:
               self.assertIsInstance(__SCREAMING_SNAKE_CASE  ,	torch.Tensor)

           # Test not batched input
           __a		 =       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
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	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							:  Dict):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)
           # create random PyTorch tensors
           __a		 =       prepare_image_inputs(self.image_processor_tester  ,	equal_resolution=__SCREAMING_SNAKE_CASE  ,	torchify=__SCREAMING_SNAKE_CASE)
           __a		 =       []
           for image in image_inputs:
               self.assertIsInstance(__SCREAMING_SNAKE_CASE  ,	torch.Tensor)
               maps.append(torch.zeros(image.shape[-2:]).long())

           # Test not batched input
           __a		 =       image_processing(image_inputs[0]  ,	maps[0]  ,	return_tensors='''pt''')
           self.assertEqual(
               encoding['''pixel_values'''].shape  ,	(
                   1,
                   self.image_processor_tester.num_channels,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(
               encoding['''labels'''].shape  ,	(
                   1,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(encoding['''labels'''].dtype  ,	torch.long)
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 255)

           # Test batched
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	__SCREAMING_SNAKE_CASE  ,	return_tensors='''pt''')
           self.assertEqual(
               encoding['''pixel_values'''].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'''],
               )  ,	)
           self.assertEqual(
               encoding['''labels'''].shape  ,	(
                   self.image_processor_tester.batch_size,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(encoding['''labels'''].dtype  ,	torch.long)
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 255)

           # Test not batched input (PIL images)
           __a ,     __a		 =       prepare_semantic_single_inputs()

           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	__SCREAMING_SNAKE_CASE  ,	return_tensors='''pt''')
           self.assertEqual(
               encoding['''pixel_values'''].shape  ,	(
                   1,
                   self.image_processor_tester.num_channels,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(
               encoding['''labels'''].shape  ,	(
                   1,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(encoding['''labels'''].dtype  ,	torch.long)
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 255)

           # Test batched input (PIL images)
           __a ,     __a		 =       prepare_semantic_batch_inputs()

           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	__SCREAMING_SNAKE_CASE  ,	return_tensors='''pt''')
           self.assertEqual(
               encoding['''pixel_values'''].shape  ,	(
                   2,
                   self.image_processor_tester.num_channels,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(
               encoding['''labels'''].shape  ,	(
                   2,
                   self.image_processor_tester.crop_size['''height'''],
                   self.image_processor_tester.crop_size['''width'''],
               )  ,	)
           self.assertEqual(encoding['''labels'''].dtype  ,	torch.long)
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 255)





       def 	_lowerCamelCase    (	self							:  Dict):


           '''simple docstring'''

           __a		 =       self.image_processing_class(**self.image_processor_dict)

           # ADE20k has 150 classes, and the background is included, so labels should be between 0 and 150
           __a ,     __a		 =       prepare_semantic_single_inputs()
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	__SCREAMING_SNAKE_CASE  ,	return_tensors='''pt''')
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 150)

           __a		 =       True
           __a		 =       image_processing(__SCREAMING_SNAKE_CASE  ,	__SCREAMING_SNAKE_CASE  ,	return_tensors='''pt''')
           self.assertTrue(encoding['''labels'''].min().item() >= 0)
           self.assertTrue(encoding['''labels'''].max().item() <= 255)





