



import tempfile
import unittest

from transformers import SPIECE_UNDERLINE, BatchEncoding, PLBartTokenizer, is_torch_available
from transformers.testing_utils import (
    get_tests_dir,
    nested_simplify,
    require_sentencepiece,
    require_tokenizers,
    require_torch,
)

from ...test_tokenization_common import TokenizerTesterMixin


a__      :				Dict   	=			get_tests_dir("fixtures/test_sentencepiece.model")


if is_torch_available():
       from transformers.models.plbart.modeling_plbart import shift_tokens_right

a__      :				int   	=			5_00_03
a__      :				Dict   	=			5_00_02



@require_sentencepiece
@require_tokenizers
class       UpperCAmelCase__(							lowerCamelCase     ,							unittest.TestCase	):

   '''simple docstring'''




   A    :  Any       				=    PLBartTokenizer
   A    :  str       				=    None
   A    :  str       				=    False


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				Tuple:




         """simple docstring"""

         super().setUp()

         # We have a SentencePiece fixture for testing
         lowercase__     =  PLBartTokenizer(lowerCAmelCase			,				language_codes='base'			,				keep_accents=lowerCAmelCase)
         tokenizer.save_pretrained(self.tmpdirname)


   def 			UpperCAmelCase		(						self  :			Optional[Any])	->				Any:




         """simple docstring"""

         lowercase__     =  PLBartTokenizer(lowerCAmelCase			,				language_codes='base'			,				keep_accents=lowerCAmelCase)

         lowercase__     =  tokenizer.tokenize('This is a test')
         self.assertListEqual(lowerCAmelCase			,				['▁This', '▁is', '▁a', '▁t', 'est'])

         self.assertListEqual(
             tokenizer.convert_tokens_to_ids(lowerCAmelCase)			,				[value + tokenizer.fairseq_offset for value in [2_85, 46, 10, 1_70, 3_82]]			,				)

         lowercase__     =  tokenizer.tokenize('I was born in 92000, and this is falsé.')
         self.assertListEqual(
             lowerCAmelCase			,				[
                 SPIECE_UNDERLINE + 'I',
                 SPIECE_UNDERLINE + 'was',
                 SPIECE_UNDERLINE + 'b',
                 'or',
                 'n',
                 SPIECE_UNDERLINE + 'in',
                 SPIECE_UNDERLINE + '',
                 '9',
                 '2',
                 '0',
                 '0',
                 '0',
                 ',',
                 SPIECE_UNDERLINE + 'and',
                 SPIECE_UNDERLINE + 'this',
                 SPIECE_UNDERLINE + 'is',
                 SPIECE_UNDERLINE + 'f',
                 'al',
                 's',
                 'é',
                 '.',
             ]			,				)
         lowercase__     =  tokenizer.convert_tokens_to_ids(lowerCAmelCase)
         self.assertListEqual(
             lowerCAmelCase			,				[
                 value + tokenizer.fairseq_offset
                 for value in [8, 21, 84, 55, 24, 19, 7, 2, 6_02, 3_47, 3_47, 3_47, 3, 12, 66, 46, 72, 80, 6, 2, 4]
             ]			,				)

         lowercase__     =  tokenizer.convert_ids_to_tokens(lowerCAmelCase)
         self.assertListEqual(
             lowerCAmelCase			,				[
                 SPIECE_UNDERLINE + 'I',
                 SPIECE_UNDERLINE + 'was',
                 SPIECE_UNDERLINE + 'b',
                 'or',
                 'n',
                 SPIECE_UNDERLINE + 'in',
                 SPIECE_UNDERLINE + '',
                 '<unk>',
                 '2',
                 '0',
                 '0',
                 '0',
                 ',',
                 SPIECE_UNDERLINE + 'and',
                 SPIECE_UNDERLINE + 'this',
                 SPIECE_UNDERLINE + 'is',
                 SPIECE_UNDERLINE + 'f',
                 'al',
                 's',
                 '<unk>',
                 '.',
             ]			,				)

         lowercase__     =  tokenizer.vocab_size
         lowercase__     =  [tokenizer.convert_ids_to_tokens(lowerCAmelCase) for x in range(end - 4			,				lowerCAmelCase)]

         self.assertListEqual(lowerCAmelCase			,				['__java__', '__python__', '__en_XX__', '<mask>'])

         lowercase__     =  'java.lang.Exception, python.lang.Exception, javascript, php, ruby, go'
         lowercase__     =  tokenizer(lowerCAmelCase).input_ids
         self.assertEqual(
             tokenizer.decode(lowerCAmelCase			,				skip_special_tokens=lowerCAmelCase			,				clean_up_tokenization_spaces=lowerCAmelCase)			,				lowerCAmelCase			,				)



   def 			UpperCAmelCase		(						self  :			Any)	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  PLBartTokenizer(lowerCAmelCase			,				language_codes='multi'			,				keep_accents=lowerCAmelCase)

         lowercase__     =  tokenizer.tokenize('This is a test')
         self.assertListEqual(lowerCAmelCase			,				['▁This', '▁is', '▁a', '▁t', 'est'])

         self.assertListEqual(
             tokenizer.convert_tokens_to_ids(lowerCAmelCase)			,				[value + tokenizer.fairseq_offset for value in [2_85, 46, 10, 1_70, 3_82]]			,				)

         lowercase__     =  tokenizer.tokenize('I was born in 92000, and this is falsé.')
         self.assertListEqual(
             lowerCAmelCase			,				[
                 SPIECE_UNDERLINE + 'I',
                 SPIECE_UNDERLINE + 'was',
                 SPIECE_UNDERLINE + 'b',
                 'or',
                 'n',
                 SPIECE_UNDERLINE + 'in',
                 SPIECE_UNDERLINE + '',
                 '9',
                 '2',
                 '0',
                 '0',
                 '0',
                 ',',
                 SPIECE_UNDERLINE + 'and',
                 SPIECE_UNDERLINE + 'this',
                 SPIECE_UNDERLINE + 'is',
                 SPIECE_UNDERLINE + 'f',
                 'al',
                 's',
                 'é',
                 '.',
             ]			,				)
         lowercase__     =  tokenizer.convert_tokens_to_ids(lowerCAmelCase)
         self.assertListEqual(
             lowerCAmelCase			,				[
                 value + tokenizer.fairseq_offset
                 for value in [8, 21, 84, 55, 24, 19, 7, 2, 6_02, 3_47, 3_47, 3_47, 3, 12, 66, 46, 72, 80, 6, 2, 4]
             ]			,				)

         lowercase__     =  tokenizer.convert_ids_to_tokens(lowerCAmelCase)
         self.assertListEqual(
             lowerCAmelCase			,				[
                 SPIECE_UNDERLINE + 'I',
                 SPIECE_UNDERLINE + 'was',
                 SPIECE_UNDERLINE + 'b',
                 'or',
                 'n',
                 SPIECE_UNDERLINE + 'in',
                 SPIECE_UNDERLINE + '',
                 '<unk>',
                 '2',
                 '0',
                 '0',
                 '0',
                 ',',
                 SPIECE_UNDERLINE + 'and',
                 SPIECE_UNDERLINE + 'this',
                 SPIECE_UNDERLINE + 'is',
                 SPIECE_UNDERLINE + 'f',
                 'al',
                 's',
                 '<unk>',
                 '.',
             ]			,				)
         lowercase__     =  tokenizer.vocab_size
         lowercase__     =  [tokenizer.convert_ids_to_tokens(lowerCAmelCase) for x in range(end - 7			,				lowerCAmelCase)]

         self.assertListEqual(
             lowerCAmelCase			,				['__java__', '__python__', '__en_XX__', '__javascript__', '__php__', '__ruby__', '__go__'])
         lowercase__     =  'java.lang.Exception, python.lang.Exception, javascript, php, ruby, go'
         lowercase__     =  tokenizer(lowerCAmelCase).input_ids
         self.assertEqual(
             tokenizer.decode(lowerCAmelCase			,				skip_special_tokens=lowerCAmelCase			,				clean_up_tokenization_spaces=lowerCAmelCase)			,				lowerCAmelCase			,				)



@require_torch
@require_sentencepiece
@require_tokenizers
class       UpperCAmelCase__(							unittest.TestCase	):

   '''simple docstring'''




   A    :  List[Any]       				=    "uclanlp/plbart-python-en_XX"
   A    :  List[Any]       				=    [
       "def maximum(a,b,c):NEW_LINE_INDENTreturn max([a,b,c])",
       "def sum(a,b,c):NEW_LINE_INDENTreturn sum([a,b,c])",
   ]
   A    :  Any       				=    [
       "Returns the maximum value of a b c.",
       "Sums the values of a b c.",
   ]
   A    :  str       				=    [
       134,
       5452,
       33460,
       33441,
       33463,
       33465,
       33463,
       33449,
       988,
       20,
       33456,
       19,
       33456,
       771,
       39,
       4258,
       889,
       3318,
       33441,
       33463,
       33465,
       33463,
       33449,
       2471,
       2,
       PYTHON_CODE,
   ]


   @classmethod
   def 			UpperCAmelCase		(						cls  :			Any)	->				str:




         """simple docstring"""

         lowercase__     =  PLBartTokenizer.from_pretrained(
             cls.checkpoint_name			,				language_codes='base'			,				src_lang='python'			,				tgt_lang='en_XX')
         lowercase__     =  1
         return cls


   def 			UpperCAmelCase		(						self  :			str)	->				List[str]:




         """simple docstring"""

         self.assertEqual(self.tokenizer.fairseq_tokens_to_ids['__java__']			,				5_00_01)
         self.assertEqual(self.tokenizer.fairseq_tokens_to_ids['__python__']			,				5_00_02)
         self.assertEqual(self.tokenizer.fairseq_tokens_to_ids['__en_XX__']			,				5_00_03)


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  self.tokenizer.batch_encode_plus(self.src_text).input_ids[0]
         self.assertListEqual(self.expected_src_tokens			,				lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			Dict)	->				Dict:




         """simple docstring"""

         self.assertIn(lowerCAmelCase			,				self.tokenizer.all_special_ids)
         lowercase__     =  [EN_CODE, 90_37, 3_34_42, 57, 7_52, 1_53, 14, 56, 18, 9, 2]
         lowercase__     =  self.tokenizer.decode(lowerCAmelCase			,				skip_special_tokens=lowerCAmelCase)
         lowercase__     =  self.tokenizer.decode(generated_ids[1:]			,				skip_special_tokens=lowerCAmelCase)
         self.assertEqual(lowerCAmelCase			,				lowerCAmelCase)
         self.assertNotIn(self.tokenizer.eos_token			,				lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			Any)	->				int:




         """simple docstring"""

         lowercase__     =  ['def sum(a,b,c):NEW_LINE_INDENTreturn sum([a,b,c])' * 20]
         self.assertIsInstance(src_text[0]			,				lowerCAmelCase)
         lowercase__     =  10
         lowercase__     =  self.tokenizer(lowerCAmelCase			,				max_length=lowerCAmelCase			,				truncation=lowerCAmelCase).input_ids[0]
         self.assertEqual(ids[-2]			,				2)
         self.assertEqual(ids[-1]			,				lowerCAmelCase)
         self.assertEqual(len(lowerCAmelCase)			,				lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			Dict)	->				Optional[Any]:




         """simple docstring"""

         self.assertListEqual(self.tokenizer.convert_tokens_to_ids(['<mask>', '__java__'])			,				[5_00_04, 5_00_01])


   def 			UpperCAmelCase		(						self  :			Optional[Any])	->				Tuple:




         """simple docstring"""

         lowercase__     =  tempfile.mkdtemp()
         lowercase__     =  self.tokenizer.fairseq_tokens_to_ids
         self.tokenizer.save_pretrained(lowerCAmelCase)
         lowercase__     =  PLBartTokenizer.from_pretrained(lowerCAmelCase)
         self.assertDictEqual(new_tok.fairseq_tokens_to_ids			,				lowerCAmelCase)


   @require_torch
   def 			UpperCAmelCase		(						self  :			Any)	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  self.tokenizer(self.src_text			,				text_target=self.tgt_text			,				padding=lowerCAmelCase			,				return_tensors='pt')
         lowercase__     =  shift_tokens_right(batch['labels']			,				self.tokenizer.pad_token_id)

         # fairseq batch: https://gist.github.com/sshleifer/cba08bc2109361a74ac3760a7e30e4f4
         self.assertEqual(batch.input_ids[1][-2:].tolist()			,				[2, PYTHON_CODE])
         self.assertEqual(batch.decoder_input_ids[1][0]			,				lowerCAmelCase)
         self.assertEqual(batch.decoder_input_ids[1][-1]			,				2)
         self.assertEqual(batch.labels[1][-2:].tolist()			,				[2, EN_CODE])


   @require_torch
   def 			UpperCAmelCase		(						self  :			Dict)	->				str:




         """simple docstring"""

         lowercase__     =  self.tokenizer(
             self.src_text			,				text_target=self.tgt_text			,				padding=lowerCAmelCase			,				truncation=lowerCAmelCase			,				max_length=len(self.expected_src_tokens)			,				return_tensors='pt'			,				)
         lowercase__     =  shift_tokens_right(batch['labels']			,				self.tokenizer.pad_token_id)

         self.assertIsInstance(lowerCAmelCase			,				lowerCAmelCase)

         self.assertEqual((2, 26)			,				batch.input_ids.shape)
         self.assertEqual((2, 26)			,				batch.attention_mask.shape)
         lowercase__     =  batch.input_ids.tolist()[0]
         self.assertListEqual(self.expected_src_tokens			,				lowerCAmelCase)
         self.assertEqual(2			,				batch.decoder_input_ids[0, -1])  # EOS
         # Test that special tokens are reset
         self.assertEqual(self.tokenizer.prefix_tokens			,				[])
         self.assertEqual(self.tokenizer.suffix_tokens			,				[self.tokenizer.eos_token_id, PYTHON_CODE])


   def 			UpperCAmelCase		(						self  :			int)	->				str:




         """simple docstring"""

         lowercase__     =  self.tokenizer(self.src_text			,				padding=lowerCAmelCase			,				truncation=lowerCAmelCase			,				max_length=3			,				return_tensors='pt')
         lowercase__     =  self.tokenizer(
             text_target=self.tgt_text			,				padding=lowerCAmelCase			,				truncation=lowerCAmelCase			,				max_length=10			,				return_tensors='pt')
         lowercase__     =  targets['input_ids']
         lowercase__     =  shift_tokens_right(lowerCAmelCase			,				self.tokenizer.pad_token_id)

         self.assertEqual(batch.input_ids.shape[1]			,				3)
         self.assertEqual(batch.decoder_input_ids.shape[1]			,				10)



   @require_torch
   def 			UpperCAmelCase		(						self  :			str)	->				Tuple:




         """simple docstring"""

         lowercase__     =  self.tokenizer._build_translation_inputs(
             'A test'			,				return_tensors='pt'			,				src_lang='en_XX'			,				tgt_lang='java')

         self.assertEqual(
             nested_simplify(lowerCAmelCase)			,				{
                 # A, test, EOS, en_XX
                 'input_ids': [[1_50, 2_42, 2, 5_00_03]],
                 'attention_mask': [[1, 1, 1, 1]],
                 # java
                 'forced_bos_token_id': 5_00_01,
             }			,				)





from __future__ import annotations

import unittest

from transformers import FunnelConfig, is_tf_available
from transformers.testing_utils import require_tf

from ...test_configuration_common import ConfigTester
from ...test_modeling_tf_common import TFModelTesterMixin, ids_tensor, random_attention_mask
from ...test_pipeline_mixin import PipelineTesterMixin


if is_tf_available():
       import tensorflow as tf

       from transformers import (
           TFFunnelBaseModel,
           TFFunnelForMaskedLM,
           TFFunnelForMultipleChoice,
           TFFunnelForPreTraining,
           TFFunnelForQuestionAnswering,
           TFFunnelForSequenceClassification,
           TFFunnelForTokenClassification,
           TFFunnelModel,
       )



class       UpperCAmelCase__:

   '''simple docstring'''




   def __init__(						self  :			Tuple			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			Dict=13			,				lowerCAmelCase  :			Dict=7			,				lowerCAmelCase  :			int=True			,				lowerCAmelCase  :			Optional[Any]=True			,				lowerCAmelCase  :			str=True			,				lowerCAmelCase  :			int=True			,				lowerCAmelCase  :			List[Any]=99			,				lowerCAmelCase  :			List[Any]=[1, 1, 2]			,				lowerCAmelCase  :			Optional[Any]=1			,				lowerCAmelCase  :			int=32			,				lowerCAmelCase  :			Union[str, Any]=4			,				lowerCAmelCase  :			Tuple=8			,				lowerCAmelCase  :			int=37			,				lowerCAmelCase  :			Any="gelu_new"			,				lowerCAmelCase  :			str=0.1			,				lowerCAmelCase  :			List[str]=0.1			,				lowerCAmelCase  :			Dict=0.0			,				lowerCAmelCase  :			str=5_12			,				lowerCAmelCase  :			str=3			,				lowerCAmelCase  :			List[Any]=0.02			,				lowerCAmelCase  :			Union[str, Any]=3			,				lowerCAmelCase  :			Any=4			,				lowerCAmelCase  :			List[Any]=None			,				lowerCAmelCase  :			Optional[int]=False			,				)	->				List[Any]:




         """simple docstring"""

         lowercase__     =  parent
         lowercase__     =  batch_size
         lowercase__     =  seq_length
         lowercase__     =  is_training
         lowercase__     =  use_input_mask
         lowercase__     =  use_token_type_ids
         lowercase__     =  use_labels
         lowercase__     =  vocab_size
         lowercase__     =  block_sizes
         lowercase__     =  num_decoder_layers
         lowercase__     =  d_model
         lowercase__     =  n_head
         lowercase__     =  d_head
         lowercase__     =  d_inner
         lowercase__     =  hidden_act
         lowercase__     =  hidden_dropout
         lowercase__     =  attention_dropout
         lowercase__     =  activation_dropout
         lowercase__     =  max_position_embeddings
         lowercase__     =  type_vocab_size
         lowercase__     =  2
         lowercase__     =  num_labels
         lowercase__     =  num_choices
         lowercase__     =  scope
         lowercase__     =  initializer_std

         # Used in the tests to check the size of the first attention layer
         lowercase__     =  n_head
         # Used in the tests to check the size of the first hidden state
         lowercase__     =  self.d_model
         # Used in the tests to check the number of output hidden states/attentions
         lowercase__     =  sum(self.block_sizes) + (0 if base else self.num_decoder_layers)
         # FunnelModel adds two hidden layers: input embeddings and the sum of the upsampled encoder hidden state with
         # the last hidden state of the first block (which is the first hidden state of the decoder).
         if not base:
               lowercase__     =  self.num_hidden_layers + 2


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				str:




         """simple docstring"""

         lowercase__     =  ids_tensor([self.batch_size, self.seq_length]			,				self.vocab_size)

         lowercase__     =  None
         if self.use_input_mask:
               lowercase__     =  random_attention_mask([self.batch_size, self.seq_length])

         lowercase__     =  None
         if self.use_token_type_ids:
               lowercase__     =  ids_tensor([self.batch_size, self.seq_length]			,				self.type_vocab_size)

         lowercase__     =  None
         lowercase__     =  None
         lowercase__     =  None
         if self.use_labels:
               lowercase__     =  ids_tensor([self.batch_size]			,				self.type_sequence_label_size)
               lowercase__     =  ids_tensor([self.batch_size, self.seq_length]			,				self.num_labels)
               lowercase__     =  ids_tensor([self.batch_size]			,				self.num_choices)

         lowercase__     =  FunnelConfig(
             vocab_size=self.vocab_size			,				block_sizes=self.block_sizes			,				num_decoder_layers=self.num_decoder_layers			,				d_model=self.d_model			,				n_head=self.n_head			,				d_head=self.d_head			,				d_inner=self.d_inner			,				hidden_act=self.hidden_act			,				hidden_dropout=self.hidden_dropout			,				attention_dropout=self.attention_dropout			,				activation_dropout=self.activation_dropout			,				max_position_embeddings=self.max_position_embeddings			,				type_vocab_size=self.type_vocab_size			,				initializer_std=self.initializer_std			,				)

         return (
             config,
             input_ids,
             token_type_ids,
             input_mask,
             sequence_labels,
             token_labels,
             choice_labels,
         )


   def 			UpperCAmelCase		(						self  :			Dict			,				lowerCAmelCase  :			List[Any]			,				lowerCAmelCase  :			Dict			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Union[str, Any]			,				lowerCAmelCase  :			Any			,				lowerCAmelCase  :			Optional[Any]			,				lowerCAmelCase  :			Union[str, Any]			,				)	->				int:




         """simple docstring"""

         lowercase__     =  TFFunnelModel(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)

         lowercase__     =  [input_ids, input_mask]
         lowercase__     =  model(lowerCAmelCase)

         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, self.seq_length, self.d_model))

         lowercase__     =  False
         lowercase__     =  TFFunnelModel(config=lowerCAmelCase)
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, self.seq_length, self.d_model))

         lowercase__     =  False
         lowercase__     =  TFFunnelModel(config=lowerCAmelCase)
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, self.seq_length, self.d_model))


   def 			UpperCAmelCase		(						self  :			List[str]			,				lowerCAmelCase  :			List[Any]			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			Any			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			int			,				lowerCAmelCase  :			Optional[Any]			,				)	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  TFFunnelBaseModel(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)

         lowercase__     =  [input_ids, input_mask]
         lowercase__     =  model(lowerCAmelCase)

         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, 2, self.d_model))

         lowercase__     =  False
         lowercase__     =  TFFunnelBaseModel(config=lowerCAmelCase)
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, 3, self.d_model))

         lowercase__     =  False
         lowercase__     =  TFFunnelBaseModel(config=lowerCAmelCase)
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.last_hidden_state.shape			,				(self.batch_size, 2, self.d_model))


   def 			UpperCAmelCase		(						self  :			Tuple			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			Optional[Any]			,				lowerCAmelCase  :			Dict			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			Tuple			,				)	->				str:




         """simple docstring"""

         lowercase__     =  TFFunnelForPreTraining(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.logits.shape			,				(self.batch_size, self.seq_length))


   def 			UpperCAmelCase		(						self  :			List[Any]			,				lowerCAmelCase  :			Any			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			Union[str, Any]			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			Optional[Any]			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Tuple			,				)	->				Optional[int]:




         """simple docstring"""

         lowercase__     =  TFFunnelForMaskedLM(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.logits.shape			,				(self.batch_size, self.seq_length, self.vocab_size))


   def 			UpperCAmelCase		(						self  :			Union[str, Any]			,				lowerCAmelCase  :			int			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Union[str, Any]			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			Dict			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			List[Any]			,				)	->				Optional[int]:




         """simple docstring"""

         lowercase__     =  self.num_labels
         lowercase__     =  TFFunnelForSequenceClassification(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.logits.shape			,				(self.batch_size, self.num_labels))


   def 			UpperCAmelCase		(						self  :			List[Any]			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			int			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			Union[str, Any]			,				lowerCAmelCase  :			List[Any]			,				)	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  self.num_choices
         lowercase__     =  TFFunnelForMultipleChoice(config=lowerCAmelCase)
         lowercase__     =  tf.tile(tf.expand_dims(lowerCAmelCase			,				1)			,				(1, self.num_choices, 1))
         lowercase__     =  tf.tile(tf.expand_dims(lowerCAmelCase			,				1)			,				(1, self.num_choices, 1))
         lowercase__     =  tf.tile(tf.expand_dims(lowerCAmelCase			,				1)			,				(1, self.num_choices, 1))
         lowercase__     =  {
             'input_ids': multiple_choice_inputs_ids,
             'attention_mask': multiple_choice_input_mask,
             'token_type_ids': multiple_choice_token_type_ids,
         }
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.logits.shape			,				(self.batch_size, self.num_choices))


   def 			UpperCAmelCase		(						self  :			List[str]			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			List[Any]			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Union[str, Any]			,				lowerCAmelCase  :			int			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			Any			,				)	->				Union[str, Any]:




         """simple docstring"""

         lowercase__     =  self.num_labels
         lowercase__     =  TFFunnelForTokenClassification(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.logits.shape			,				(self.batch_size, self.seq_length, self.num_labels))


   def 			UpperCAmelCase		(						self  :			int			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			int			,				lowerCAmelCase  :			Optional[int]			,				lowerCAmelCase  :			List[str]			,				lowerCAmelCase  :			str			,				lowerCAmelCase  :			Tuple			,				lowerCAmelCase  :			Tuple			,				)	->				Optional[int]:




         """simple docstring"""

         lowercase__     =  TFFunnelForQuestionAnswering(config=lowerCAmelCase)
         lowercase__     =  {'input_ids': input_ids, 'attention_mask': input_mask, 'token_type_ids': token_type_ids}
         lowercase__     =  model(lowerCAmelCase)
         self.parent.assertEqual(result.start_logits.shape			,				(self.batch_size, self.seq_length))
         self.parent.assertEqual(result.end_logits.shape			,				(self.batch_size, self.seq_length))



   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				str:




         """simple docstring"""

         lowercase__     =  self.prepare_config_and_inputs()
         (
             (
             lowercase__
         ),	(
             lowercase__
         ),	(
             lowercase__
         ),	(
             lowercase__
         ),	(
             lowercase__
         ),	(
             lowercase__
         ),	(
             lowercase__
         ),	
         )     =  config_and_inputs
         lowercase__     =  {'input_ids': input_ids, 'token_type_ids': token_type_ids, 'attention_mask': input_mask}
         return config, inputs_dict



@require_tf
class       UpperCAmelCase__(							lowerCamelCase     ,							lowerCamelCase     ,							unittest.TestCase	):

   '''simple docstring'''




   A    :  int       				=    (
       (
           TFFunnelModel,
           TFFunnelForMaskedLM,
           TFFunnelForPreTraining,
           TFFunnelForQuestionAnswering,
           TFFunnelForTokenClassification,
       )
       if is_tf_available()
       else ()
   )
   A    :  Dict       				=    (
       {
           "feature-extraction": (TFFunnelBaseModel, TFFunnelModel),
           "fill-mask": TFFunnelForMaskedLM,
           "question-answering": TFFunnelForQuestionAnswering,
           "text-classification": TFFunnelForSequenceClassification,
           "token-classification": TFFunnelForTokenClassification,
           "zero-shot": TFFunnelForSequenceClassification,
       }
       if is_tf_available()
       else {}
   )
   A    :  Optional[int]       				=    False
   A    :  Optional[int]       				=    False


   def 			UpperCAmelCase		(						self  :			Tuple)	->				str:




         """simple docstring"""

         lowercase__     =  TFFunnelModelTester(self)
         lowercase__     =  ConfigTester(self			,				config_class=lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			int)	->				Optional[int]:




         """simple docstring"""

         self.config_tester.run_common_tests()


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				Union[str, Any]:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_model(*lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				Union[str, Any]:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_pretraining(*lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			int)	->				Union[str, Any]:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_masked_lm(*lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			List[str])	->				Union[str, Any]:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_token_classification(*lowerCAmelCase)



   def 			UpperCAmelCase		(						self  :			Dict)	->				Dict:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_question_answering(*lowerCAmelCase)



@require_tf
class       UpperCAmelCase__(							lowerCamelCase     ,							unittest.TestCase	):

   '''simple docstring'''




   A    :  Tuple       				=    (
       (TFFunnelBaseModel, TFFunnelForMultipleChoice, TFFunnelForSequenceClassification) if is_tf_available() else ()
   )
   A    :  List[str]       				=    False
   A    :  int       				=    False


   def 			UpperCAmelCase		(						self  :			Any)	->				List[Any]:




         """simple docstring"""

         lowercase__     =  TFFunnelModelTester(self			,				base=lowerCAmelCase)
         lowercase__     =  ConfigTester(self			,				config_class=lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			Union[str, Any])	->				Optional[int]:




         """simple docstring"""

         self.config_tester.run_common_tests()


   def 			UpperCAmelCase		(						self  :			Tuple)	->				int:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_base_model(*lowerCAmelCase)


   def 			UpperCAmelCase		(						self  :			int)	->				str:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_sequence_classification(*lowerCAmelCase)



   def 			UpperCAmelCase		(						self  :			List[str])	->				Optional[Any]:




         """simple docstring"""

         lowercase__     =  self.model_tester.prepare_config_and_inputs()
         self.model_tester.create_and_check_for_multiple_choice(*lowerCAmelCase)
