"""simple docstring"""


def 					__a   (		A    ,				A      ):






     '''simple docstring'''



     return int((input_a, input_a).count(0      ) == 0      )







def 					__a   (		):






     '''simple docstring'''



     assert and_gate(0    ,				0      ) == 0
     assert and_gate(0    ,				1      ) == 0
     assert and_gate(1    ,				0      ) == 0
     assert and_gate(1    ,				1      ) == 1


if __name__ == "__main__":
 test_and_gate()
 print(and_gate(1, 0))
 print(and_gate(0, 0))
 print(and_gate(0, 1))
 print(and_gate(1, 1))





"""simple docstring"""


import unittest

from transformers import AutoConfig, AutoTokenizer, BertConfig, TensorType, is_flax_available
from transformers.testing_utils import DUMMY_UNKNOWN_IDENTIFIER, require_flax, slow


if is_flax_available():
 import jax

 from transformers.models.auto.modeling_flax_auto import FlaxAutoModel
 from transformers.models.bert.modeling_flax_bert import FlaxBertModel
 from transformers.models.roberta.modeling_flax_roberta import FlaxRobertaModel






@require_flax
class    a__    (   unittest.TestCase	):


   @slow
   def 	snake_case__	(     self  ):
        '''simple docstring'''



        for model_name in ["bert-base-cased", "bert-large-uncased"]:
             with self.subTest(_UpperCAmelCase  ):
                  lowercase__				 =							AutoConfig.from_pretrained(_UpperCAmelCase  )
                  self.assertIsNotNone(_UpperCAmelCase  )
                  self.assertIsInstance(_UpperCAmelCase,				_UpperCAmelCase  )

                  lowercase__				 =							FlaxAutoModel.from_pretrained(_UpperCAmelCase  )
                  self.assertIsNotNone(_UpperCAmelCase  )
                  self.assertIsInstance(_UpperCAmelCase,				_UpperCAmelCase  )


   @slow
   def 	snake_case__	(     self  ):
        '''simple docstring'''



        for model_name in ["roberta-base", "roberta-large"]:
             with self.subTest(_UpperCAmelCase  ):
                  lowercase__				 =							AutoConfig.from_pretrained(_UpperCAmelCase  )
                  self.assertIsNotNone(_UpperCAmelCase  )
                  self.assertIsInstance(_UpperCAmelCase,				_UpperCAmelCase  )

                  lowercase__				 =							FlaxAutoModel.from_pretrained(_UpperCAmelCase  )
                  self.assertIsNotNone(_UpperCAmelCase  )
                  self.assertIsInstance(_UpperCAmelCase,				_UpperCAmelCase  )


   @slow
   def 	snake_case__	(     self  ):
        '''simple docstring'''



        for model_name in ["bert-base-cased", "bert-large-uncased"]:
             lowercase__				 =							AutoTokenizer.from_pretrained(_UpperCAmelCase  )
             lowercase__				 =							FlaxBertModel.from_pretrained(_UpperCAmelCase  )
             lowercase__				 =							tokenizer("Do you support jax jitted function?",				return_tensors=TensorType.JAX  )

             @jax.jit
             def eval(**_UpperCAmelCase  ):
                  return model(**_UpperCAmelCase  )

             eval(**_UpperCAmelCase  ).block_until_ready()


   @slow
   def 	snake_case__	(     self  ):
        '''simple docstring'''



        for model_name in ["roberta-base", "roberta-large"]:
             lowercase__				 =							AutoTokenizer.from_pretrained(_UpperCAmelCase  )
             lowercase__				 =							FlaxRobertaModel.from_pretrained(_UpperCAmelCase  )
             lowercase__				 =							tokenizer("Do you support jax jitted function?",				return_tensors=TensorType.JAX  )

             @jax.jit
             def eval(**_UpperCAmelCase  ):
                  return model(**_UpperCAmelCase  )

             eval(**_UpperCAmelCase  ).block_until_ready()


   def 	snake_case__	(     self  ):
        '''simple docstring'''



        with self.assertRaisesRegex(
            _UpperCAmelCase,				"bert-base is not a local folder and is not a valid model identifier"  ):
             lowercase__				 =							FlaxAutoModel.from_pretrained("bert-base"  )


   def 	snake_case__	(     self  ):
        '''simple docstring'''



        with self.assertRaisesRegex(
            _UpperCAmelCase,				R"aaaaaa is not a valid git identifier \(branch name, tag name or commit id\)"  ):
             lowercase__				 =							FlaxAutoModel.from_pretrained(_UpperCAmelCase,				revision="aaaaaa"  )


   def 	snake_case__	(     self  ):
        '''simple docstring'''



        with self.assertRaisesRegex(
            _UpperCAmelCase,				"hf-internal-testing/config-no-model does not appear to have a file named flax_model.msgpack",				):
             lowercase__				 =							FlaxAutoModel.from_pretrained("hf-internal-testing/config-no-model"  )


   def 	snake_case__	(     self  ):
        '''simple docstring'''



        with self.assertRaisesRegex(_UpperCAmelCase,				"Use `from_pt=True` to load this model"  ):
             lowercase__				 =							FlaxAutoModel.from_pretrained("hf-internal-testing/tiny-bert-pt-only"  )




