



"""simple docstring"""







from typing import List

import PIL
import torch
from PIL import Image

from ...configuration_utils import ConfigMixin
from ...models.modeling_utils import ModelMixin
from ...utils import PIL_INTERPOLATION

class 			_a     (   SCREAMING_SNAKE_CASE__    ,  SCREAMING_SNAKE_CASE__):



			def __init__( self       :							Union[str, Any]		)       ->				List[Any]:
									super().__init__()

									self.register_buffer("watermark_image"      ,  torch.zeros((62, 62, 4)		)		)
									snake_case      : Optional[int]							=   None



			def 							__lowercase				( self       :							Optional[int]      ,  _lowercase       :							List[PIL.Image.Image]      ,  _lowercase       :							Optional[Any]=None		)       ->				List[Any]:
									# copied from https://github.com/deep-floyd/IF/blob/b77482e36ca2031cb94dbca1001fc1e6400bf4ab/deepfloyd_if/modules/base.py#L287

									snake_case      : str							=   images[0].height
									snake_case      : int							=   images[0].width

									snake_case      : Union[str, Any]							=   sample_size or h

									snake_case      : List[str]							=   min(h / sample_size      ,  w / sample_size		)
									snake_case						, snake_case      : Optional[int]							=   (int(h / coef		), int(w / coef		)) if coef < 1 else (h, w)

									snake_case						, snake_case      : List[Any]							=   1024**2, img_w * img_h
									snake_case      : List[str]							=   (Sa / Sa) ** 0.5
									snake_case						, snake_case						, snake_case      : int							=   int(K * 62		), img_w - int(14 * K		), img_h - int(14 * K		)

									if self.watermark_image_as_pil is None:
															snake_case      : Union[str, Any]							=   self.watermark_image.to(torch.uinta		).cpu().numpy()
															snake_case      : List[Any]							=   Image.fromarray(_lowercase      ,  mode="RGBA"		)
															snake_case      : str							=   watermark_image

									snake_case      : List[Any]							=   self.watermark_image_as_pil.resize(
									    (wm_size, wm_size)      ,  PIL_INTERPOLATION["bicubic"]      ,  reducing_gap=_lowercase		)

									for pil_img in images:
															pil_img.paste(_lowercase      ,  box=(wm_x - wm_size, wm_y - wm_size, wm_x, wm_y)      ,  mask=wm_img.split()[-1]		)

									return images








"""simple docstring"""







import unittest

from cer import CER


A    =	CER()

class 			_a     (   unittest.TestCase):



			def 							__lowercase				( self       :							Optional[Any]		)       ->				Tuple:
									snake_case      : Union[str, Any]							=   ["White House"]
									snake_case      : Tuple							=   ["white house"]
									# S = 2, D = 0, I = 0, N = 11, CER = 2 / 11
									snake_case      : int							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.1818181818		) < 1E-6		)



			def 							__lowercase				( self       :							Optional[Any]		)       ->				Dict:
									snake_case      : Tuple							=   ["were wolf"]
									snake_case      : Union[str, Any]							=   ["werewolf"]
									# S = 0, D = 0, I = 1, N = 9, CER = 1 / 9
									snake_case      : List[str]							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.1111111		) < 1E-6		)

									snake_case      : List[str]							=   ["werewolf"]
									snake_case      : Optional[Any]							=   ["weae     wolf"]
									# S = 1, D = 1, I = 0, N = 8, CER = 0.25
									snake_case      : Optional[int]							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.25		) < 1E-6		)

									# consecutive whitespaces case 1
									snake_case      : Optional[int]							=   ["were wolf"]
									snake_case      : Dict							=   ["were               wolf"]
									# S = 0, D = 0, I = 0, N = 9, CER = 0
									snake_case      : Dict							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.0		) < 1E-6		)

									# consecutive whitespaces case 2
									snake_case      : Dict							=   ["were   wolf"]
									snake_case      : Dict							=   ["were               wolf"]
									# S = 0, D = 0, I = 0, N = 9, CER = 0
									snake_case      : Optional[int]							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.0		) < 1E-6		)



			def 							__lowercase				( self       :							List[Any]		)       ->				Union[str, Any]:
									snake_case      : str							=   ["werewolf"]
									snake_case      : List[Any]							=   ["weaewolf"]
									# S = 1, D = 0, I = 0, N = 8, CER = 0.125
									snake_case      : str							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.125		) < 1E-6		)



			def 							__lowercase				( self       :							int		)       ->				int:
									snake_case      : Dict							=   ["werewolf"]
									snake_case      : Optional[Any]							=   ["wereawolf"]
									# S = 0, D = 1, I = 0, N = 8, CER = 0.125
									snake_case      : List[str]							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.125		) < 1E-6		)



			def 							__lowercase				( self       :							Union[str, Any]		)       ->				List[Any]:
									snake_case      : Tuple							=   ["werewolf"]
									snake_case      : Union[str, Any]							=   ["wereolf"]
									# S = 0, D = 0, I = 1, N = 8, CER = 0.125
									snake_case      : str							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.125		) < 1E-6		)



			def 							__lowercase				( self       :							int		)       ->				List[str]:
									snake_case      : int							=   ["werewolf"]
									snake_case      : Optional[Any]							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertEqual(_lowercase      ,  0.0		)



			def 							__lowercase				( self       :							List[Any]		)       ->				Optional[Any]:
									snake_case      : str							=   ["werewolf", "I am your father"]
									snake_case      : str							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertEqual(_lowercase      ,  0.0		)

									snake_case      : Union[str, Any]							=   ["werewolf", "I am your father", "doge"]
									snake_case      : str							=   ["werxwolf", "I       am your father", "doge"]
									# S = 1, D = 0, I = 0, N = 28, CER = 1 / 28
									snake_case      : Dict							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.03571428		) < 1E-6		)



			def 							__lowercase				( self       :							Any		)       ->				Union[str, Any]:
									snake_case      : Union[str, Any]							=   ["My hovercraft", "is full of eels"]
									snake_case      : List[str]							=   ["My hovercraft is full", " of eels"]
									# S = 0, D = 0, I = 2, N = 28, CER = 2 / 28
									# whitespace at the front of " of eels" will be strip during preporcessing
									# so need to insert 2 whitespaces
									snake_case      : int							=   cer.compute(predictions=_lowercase      ,  references=_lowercase      ,  concatenate_texts=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.071428		) < 1E-6		)



			def 							__lowercase				( self       :							Any		)       ->				Dict:
									snake_case      : Dict							=   ["我能吞下玻璃而不伤身体"]
									snake_case      : List[str]							=   [" 能吞虾玻璃而 不霜身体啦"]
									# S = 3, D = 2, I = 0, N = 11, CER = 5 / 11
									snake_case      : str							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.4545454545		) < 1E-6		)

									snake_case      : List[Any]							=   ["我能吞下玻璃", "而不伤身体"]
									snake_case      : int							=   ["我    能 吞 下 玻 璃", "而不伤身体"]
									# S = 0, D = 5, I = 0, N = 11, CER = 5 / 11
									snake_case      : Dict							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertTrue(abs(char_error_rate - 0.454545454545		) < 1E-6		)

									snake_case      : int							=   ["我能吞下玻璃而不伤身体"]
									snake_case      : int							=   cer.compute(predictions=_lowercase      ,  references=_lowercase		)
									self.assertFalse(_lowercase      ,  0.0		)



			def 							__lowercase				( self       :							List[Any]		)       ->				List[Any]:
									snake_case      : int							=   [""]
									snake_case      : Optional[int]							=   ["Hypothesis"]
									with self.assertRaises(_lowercase		):
															cer.compute(predictions=_lowercase      ,  references=_lowercase		)


if __name__ == "__main__":
			unittest.main()



