



from __future__ import annotations

import unittest

from transformers import is_tf_available
from transformers.testing_utils import require_sentencepiece, require_tf, require_tokenizers, slow


if is_tf_available():
						import numpy as np
						import tensorflow as tf

						from transformers import TFXLMRobertaModel






@require_tf
@require_sentencepiece
@require_tokenizers
class      UpperCamelCase    (						unittest.TestCase     ):



	@slow
	def 			__A							(       self		):
							A__							=       TFXLMRobertaModel.from_pretrained("jplu/tf-xlm-roberta-base"		)

							A__							=       {
							    "input_ids": tf.convert_to_tensor([[0, 2_646, 10_269, 83, 99_942, 2]]			,	dtype=tf.intaa		),  # "My dog is cute"
							    "attention_mask": tf.convert_to_tensor([[1, 1, 1, 1, 1, 1]]			,	dtype=tf.intaa		),
							}

							A__							=       model(UpperCAmelCase__		)["last_hidden_state"]
							A__							=       tf.TensorShape((1, 6, 768)		)
							self.assertEqual(output.shape			,	UpperCAmelCase__		)
							# compare the actual values for a slice.
							A__							=       tf.convert_to_tensor(
							    [
							        [
							            [0.0_681_762, 0.10_894_451, 0.06_772_504],
							            [-0.06_423_668, 0.02_366_615, 0.04_329_344],
							            [-0.06_057_295, 0.09_974_135, -0.00_070_584],
							        ]
							    ]			,	dtype=tf.floataa			,	)

							self.assertTrue(np.allclose(output[:, :3, :3].numpy()			,	expected_slice.numpy()			,	atol=1e-4		)		)










def  UpperCamelCase						(       _A			:				int			,							_A			:				int    )->       int:





						"""simple docstring"""





						return int((input_a, input_a).count(0    ) == 0    )
def  UpperCamelCase						(       )->       None:





						"""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))





