



'''simple docstring'''



from timeit import timeit

lowerCamelCase	 =       {
    """MALAYALAM""": True,
    """String""": False,
    """rotor""": True,
    """level""": True,
    """A""": True,
    """BB""": True,
    """ABC""": False,
    """amanaplanacanalpanama""": True,  # "a man a plan a canal panama"
}
# Ensure our test data is valid
assert all((key == key[::-1]) is value for key, value in test_data.items())



def   _A       (   _lowerCAmelCase					):



							"""simple docstring"""



							__lowercase       	=0
							__lowercase       	=len(_lowerCAmelCase					) - 1
							while start_i < end_i:
														if s[start_i] == s[end_i]:
																					start_i += 1
																					end_i -= 1
														else:
																					return False
							return True



def   _A       (   _lowerCAmelCase					):



							"""simple docstring"""



							__lowercase       	=len(_lowerCAmelCase					) // 2
							__lowercase       	=len(_lowerCAmelCase					)

							# We need to traverse till half of the length of string
							# as we can get access of the i'th last element from
							# i'th index.
							# eg: [0,1,2,3,4,5] => 4th index can be accessed
							# with the help of 1st index (i==n-i-1)
							# where n is length of string
							return all(s[i] == s[n - i - 1] for i in range(_lowerCAmelCase					)					)



def   _A       (   _lowerCAmelCase					):



							"""simple docstring"""



							if len(_lowerCAmelCase					) <= 2:
														return True
							if s[0] == s[len(_lowerCAmelCase					) - 1]:
														return is_palindrome_recursive(s[1:-1]					)
							else:
														return False



def   _A       (   _lowerCAmelCase					):



							"""simple docstring"""



							return s == s[::-1]



def   _A       (   _lowerCAmelCase					):



							"""simple docstring"""



							__lowercase       	=f"""all({name}(key) is value for key, value in test_data.items())"""
							__lowercase       	=f"""from __main__ import test_data, {name}"""
							__lowercase       	=500_000
							__lowercase       	=timeit(stmt=_lowerCAmelCase  ,   setup=_lowerCAmelCase  ,   number=_lowerCAmelCase					)
							print(f"""{name:<35} finished {number:,} runs in {result:.5f} seconds"""					)


if __name__ == "__main__":
		for key, value in test_data.items():
				assert is_palindrome(key) is is_palindrome_recursive(key)
				assert is_palindrome(key) is is_palindrome_slice(key)
				print(f"{key:21} {value}")
		print("""a man a plan a canal panama""")

		# finished 500,000 runs in 0.46793 seconds
		benchmark_function("""is_palindrome_slice""")
		# finished 500,000 runs in 0.85234 seconds
		benchmark_function("""is_palindrome""")
		# finished 500,000 runs in 1.32028 seconds
		benchmark_function("""is_palindrome_recursive""")
		# finished 500,000 runs in 2.08679 seconds
		benchmark_function("""is_palindrome_traversal""")








'''simple docstring'''



import gc
import unittest

from diffusers import FlaxStableDiffusionInpaintPipeline
from diffusers.utils import is_flax_available, load_image, slow
from diffusers.utils.testing_utils import require_flax


if is_flax_available():
		import jax
		import jax.numpy as jnp
		from flax.jax_utils import replicate
		from flax.training.common_utils import shard





@slow
@require_flax
class _UpperCamelCase		(     unittest.TestCase		):





				'''simple docstring'''





				def 				__lowerCamelCase     (							self  :  List[Any]):







											'''simple docstring'''

											super().tearDown()
											gc.collect()

				def 				__lowerCamelCase     (							self  :  List[str]):







											'''simple docstring'''

											__lowercase       	=load_image(
											    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'
											    '/sd2-inpaint/init_image.png')
											__lowercase       	=load_image(
											    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd2-inpaint/mask.png')

											__lowercase       	='xvjiarui/stable-diffusion-2-inpainting'
											__lowercase	,				__lowercase       	=FlaxStableDiffusionInpaintPipeline.from_pretrained(_lowerCAmelCase  ,       safety_checker=_lowerCAmelCase)

											__lowercase       	='Face of a yellow cat, high resolution, sitting on a park bench'

											__lowercase       	=jax.random.PRNGKey(0)
											__lowercase       	=5_0

											__lowercase       	=jax.device_count()
											__lowercase       	=num_samples * [prompt]
											__lowercase       	=num_samples * [init_image]
											__lowercase       	=num_samples * [mask_image]
											__lowercase	,				__lowercase	,				__lowercase       	=pipeline.prepare_inputs(_lowerCAmelCase  ,       _lowerCAmelCase  ,       _lowerCAmelCase)

											# shard inputs and rng
											__lowercase       	=replicate(_lowerCAmelCase)
											__lowercase       	=jax.random.split(_lowerCAmelCase  ,       jax.device_count())
											__lowercase       	=shard(_lowerCAmelCase)
											__lowercase       	=shard(_lowerCAmelCase)
											__lowercase       	=shard(_lowerCAmelCase)

											__lowercase       	=pipeline(
											    _lowerCAmelCase  ,       _lowerCAmelCase  ,       _lowerCAmelCase  ,       _lowerCAmelCase  ,       _lowerCAmelCase  ,       _lowerCAmelCase  ,       jit=_lowerCAmelCase)

											__lowercase       	=output.images.reshape(_lowerCAmelCase  ,       5_1_2  ,       5_1_2  ,       3)

											__lowercase       	=images[0, 2_5_3:2_5_6, 2_5_3:2_5_6, -1]

											__lowercase       	=jnp.asarray(jax.device_get(image_slice.flatten()))
											__lowercase       	=jnp.array(
											    [0.361_1307, 0.3764_9736, 0.375_7408, 0.3821_3953, 0.3929_5167, 0.384_1631, 0.4155_4978, 0.413_7475, 0.421_7084])
											print(f"""output_slice: {output_slice}""")
											assert jnp.abs(output_slice - expected_slice).max() < 1e-2



