
import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible



class    a_ (			unittest.TestCase    ):






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.bin""",
									    """safety_checker/model.safetensors""",
									    """vae/diffusion_pytorch_model.bin""",
									    """vae/diffusion_pytorch_model.safetensors""",
									    """text_encoder/pytorch_model.bin""",
									    """text_encoder/model.safetensors""",
									    """unet/diffusion_pytorch_model.bin""",
									    """unet/diffusion_pytorch_model.safetensors""",
									]
									self.assertTrue(is_safetensors_compatible(A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """unet/diffusion_pytorch_model.bin""",
									    """unet/diffusion_pytorch_model.safetensors""",
									]
									self.assertTrue(is_safetensors_compatible(A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.bin""",
									    """safety_checker/model.safetensors""",
									    """vae/diffusion_pytorch_model.bin""",
									    """vae/diffusion_pytorch_model.safetensors""",
									    """text_encoder/pytorch_model.bin""",
									    """text_encoder/model.safetensors""",
									    """unet/diffusion_pytorch_model.bin""",
									    # Removed: 'unet/diffusion_pytorch_model.safetensors',
									]
									self.assertFalse(is_safetensors_compatible(A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """text_encoder/pytorch_model.bin""",
									    """text_encoder/model.safetensors""",
									]
									self.assertTrue(is_safetensors_compatible(A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.bin""",
									    """safety_checker/model.safetensors""",
									    """vae/diffusion_pytorch_model.bin""",
									    """vae/diffusion_pytorch_model.safetensors""",
									    """text_encoder/pytorch_model.bin""",
									    # Removed: 'text_encoder/model.safetensors',
									    """unet/diffusion_pytorch_model.bin""",
									    """unet/diffusion_pytorch_model.safetensors""",
									]
									self.assertFalse(is_safetensors_compatible(A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.fp16.bin""",
									    """safety_checker/model.fp16.safetensors""",
									    """vae/diffusion_pytorch_model.fp16.bin""",
									    """vae/diffusion_pytorch_model.fp16.safetensors""",
									    """text_encoder/pytorch_model.fp16.bin""",
									    """text_encoder/model.fp16.safetensors""",
									    """unet/diffusion_pytorch_model.fp16.bin""",
									    """unet/diffusion_pytorch_model.fp16.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertTrue(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """unet/diffusion_pytorch_model.fp16.bin""",
									    """unet/diffusion_pytorch_model.fp16.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertTrue(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									# pass variant but use the non-variant filenames
									_lowercase					       =					[
									    """unet/diffusion_pytorch_model.bin""",
									    """unet/diffusion_pytorch_model.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertTrue(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.fp16.bin""",
									    """safety_checker/model.fp16.safetensors""",
									    """vae/diffusion_pytorch_model.fp16.bin""",
									    """vae/diffusion_pytorch_model.fp16.safetensors""",
									    """text_encoder/pytorch_model.fp16.bin""",
									    """text_encoder/model.fp16.safetensors""",
									    """unet/diffusion_pytorch_model.fp16.bin""",
									    # Removed: 'unet/diffusion_pytorch_model.fp16.safetensors',
									]
									_lowercase					       =					"""fp16"""
									self.assertFalse(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """text_encoder/pytorch_model.fp16.bin""",
									    """text_encoder/model.fp16.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertTrue(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									# pass variant but use the non-variant filenames
									_lowercase					       =					[
									    """text_encoder/pytorch_model.bin""",
									    """text_encoder/model.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertTrue(is_safetensors_compatible(A_				,			variant=A_  )  )






			def    UpperCamelCase_				(      self  ):
									_lowercase					       =					[
									    """safety_checker/pytorch_model.fp16.bin""",
									    """safety_checker/model.fp16.safetensors""",
									    """vae/diffusion_pytorch_model.fp16.bin""",
									    """vae/diffusion_pytorch_model.fp16.safetensors""",
									    """text_encoder/pytorch_model.fp16.bin""",
									    # 'text_encoder/model.fp16.safetensors',
									    """unet/diffusion_pytorch_model.fp16.bin""",
									    """unet/diffusion_pytorch_model.fp16.safetensors""",
									]
									_lowercase					       =					"""fp16"""
									self.assertFalse(is_safetensors_compatible(A_				,			variant=A_  )  )

"""simple docstring"""


from maths.prime_check import is_prime






def    _snake_case					(		snake_case__ :  int  ):
	if not isinstance(snake_case__						,   snake_case__  ):
		A     = F'Input value of [number={number}] must be an integer'
		raise TypeError(snake_case__  )
	if is_prime(snake_case__  ) and is_prime(number + 2  ):
		return number + 2
	else:
		return -1


if __name__ == "__main__":
		import doctest

		doctest.testmod()