






'''simple docstring'''
import unittest

from transformers import SpeechaTextaConfig
from transformers.testing_utils import is_torch_available, require_torch, torch_device

from ...generation.test_utils import GenerationTesterMixin
from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, ids_tensor
from ...test_pipeline_mixin import PipelineTesterMixin


if is_torch_available():
 import torch

 from transformers.models.speech_to_text_a.modeling_speech_to_text_a import (
     SpeechaTextaDecoder,
     SpeechaTextaForCausalLM,
 )


@require_torch
class     _UpperCAmelCase  :




    def __init__(     self					:    Union[str, Any]					,       a					:    str					,       a					:    Tuple=9_9					,       a					:    List[Any]=1_3					,       a					:    Dict=1_6					,       a					:    Dict=7					,       a					:    Tuple=True					,       a					:    Optional[Any]=True					,       a					:    List[Any]=True					,       a					:    str=False					,       a					:    Tuple=True					,       a					:    Optional[int]=2					,       a					:    Optional[Any]=3_2					,       a					:    Dict=4					,       a					:    List[str]=4					,       a					:    str=3_0					,       a					:    Optional[Any]=0					,       a					:    Dict=1					,       a					:    Optional[Any]=2					,       a					:    List[str]=None					,       ):






       '''simple docstring'''






       lowercase_		:      List[str]          =       parent
       lowercase_		:      Union[str, Any]          =       batch_size
       lowercase_		:      str          =       decoder_seq_length
       # For common tests
       lowercase_		:      Any          =       self.decoder_seq_length
       lowercase_		:      Union[str, Any]          =       is_training
       lowercase_		:      str          =       use_attention_mask
       lowercase_		:      str          =       use_labels

       lowercase_		:      int          =       vocab_size
       lowercase_		:      Union[str, Any]          =       d_model
       lowercase_		:      Optional[int]          =       d_model
       lowercase_		:      Optional[Any]          =       decoder_layers
       lowercase_		:      Optional[int]          =       decoder_layers
       lowercase_		:      Optional[int]          =       decoder_ffn_dim
       lowercase_		:      Dict          =       decoder_attention_heads
       lowercase_		:      List[str]          =       decoder_attention_heads
       lowercase_		:      Union[str, Any]          =       eos_token_id
       lowercase_		:      str          =       bos_token_id
       lowercase_		:      Tuple          =       pad_token_id
       lowercase_		:      Optional[Any]          =       decoder_start_token_id
       lowercase_		:      Dict          =       use_cache
       lowercase_		:      Optional[int]          =       max_position_embeddings

       lowercase_		:      Union[str, Any]          =       None
       lowercase_		:      List[Any]          =       decoder_seq_length
       lowercase_		:      Optional[int]          =       2
       lowercase_		:      Optional[Any]          =       1




    def 			lowerCAmelCase__			(     self					:    Union[str, Any]     ):






       '''simple docstring'''






       lowercase_		:      int          =       ids_tensor([self.batch_size, self.decoder_seq_length]					,       self.vocab_size     )

       lowercase_		:      Optional[Any]          =       None
       if self.use_attention_mask:
          lowercase_		:      int          =       ids_tensor([self.batch_size, self.decoder_seq_length]					,       vocab_size=2     )

       lowercase_		:      str          =       None
       if self.use_labels:
          lowercase_		:      List[Any]          =       ids_tensor([self.batch_size, self.decoder_seq_length]					,       self.vocab_size     )

       lowercase_		:      str          =       SpeechaTextaConfig(
           vocab_size=self.vocab_size					,       d_model=self.d_model					,       decoder_layers=self.decoder_layers					,       decoder_ffn_dim=self.decoder_ffn_dim					,       decoder_attention_heads=self.decoder_attention_heads					,       eos_token_id=self.eos_token_id					,       bos_token_id=self.bos_token_id					,       use_cache=self.use_cache					,       pad_token_id=self.pad_token_id					,       decoder_start_token_id=self.decoder_start_token_id					,       max_position_embeddings=self.max_position_embeddings					,       )

       return (
           config,
           input_ids,
           attention_mask,
           lm_labels,
       )




    def 			lowerCAmelCase__			(     self					:    int					,       a					:    Any					,       a					:    Any					,       a					:    Optional[int]					,       a					:    Optional[int]					,       ):






       '''simple docstring'''






       lowercase_		:      Optional[Any]          =       True
       lowercase_		:      List[str]          =       SpeechaTextaDecoder(config=a     ).to(a     ).eval()
       lowercase_		:      str          =       input_ids[:2]

       input_ids[input_ids == 0] += 1
       # first forward pass
       lowercase_		:      Any          =       model(a					,       use_cache=a     )
       lowercase_		:      int          =       model(a     )
       lowercase_		:      str          =       model(a					,       use_cache=a     )

       self.parent.assertTrue(len(a     ) == len(a     )     )
       self.parent.assertTrue(len(a     ) == len(a     ) + 1     )

       lowercase_		:      Union[str, Any]          =       outputs["past_key_values"]

       # create hypothetical next token and extent to next_input_ids
       lowercase_		:      Dict          =       ids_tensor((2, 1)					,       config.vocab_size - 1     ) + 1

       # append to next input_ids and
       lowercase_		:      List[str]          =       torch.cat([input_ids, next_tokens]					,       dim=-1     )

       print(a     )

       lowercase_		:      Optional[Any]          =       model(a     )["last_hidden_state"]
       lowercase_		:      int          =       model(a					,       past_key_values=a     )["last_hidden_state"]

       # select random slice
       lowercase_		:      Tuple          =       ids_tensor((1,)					,       output_from_past.shape[-1]     ).item()
       lowercase_		:      Optional[int]          =       output_from_no_past[:, next_input_ids.shape[-1] - 1, random_slice_idx].detach()
       lowercase_		:      Optional[Any]          =       output_from_past[:, 0, random_slice_idx].detach()

       # test that outputs are equal for slice
       assert torch.allclose(a					,       a					,       atol=1e-3     )





    def 			lowerCAmelCase__			(     self					:    Optional[Any]     ):






       '''simple docstring'''






       lowercase_		:      Tuple          =       self.prepare_config_and_inputs()
       (
           (
           lowercase_
       )      ,						(
           lowercase_
       )      ,						(
           lowercase_
       )      ,						(
           lowercase_
       )      ,						
       )		:      List[Any]          =       config_and_inputs

       lowercase_		:      Union[str, Any]          =       {
           "input_ids": input_ids,
           "attention_mask": attention_mask,
       }
       return config, inputs_dict


@require_torch
class     _UpperCAmelCase  (						snake_case			,   snake_case			,   snake_case			,   unittest.TestCase     ):
    __lowerCamelCase:				Dict						    =      (SpeechaTextaDecoder, SpeechaTextaForCausalLM) if is_torch_available() else ()
    __lowerCamelCase:				Optional[int]						    =      (SpeechaTextaForCausalLM,) if is_torch_available() else ()
    __lowerCamelCase:				Union[str, Any]						    =      {'text-generation': SpeechaTextaForCausalLM} if is_torch_available() else {}
    __lowerCamelCase:				int						    =      True
    __lowerCamelCase:				Optional[Any]						    =      False




    def 			lowerCAmelCase__			(     self					:    str					,       ):






       '''simple docstring'''






       lowercase_		:      Union[str, Any]          =       SpeechaTextaStandaloneDecoderModelTester(self					,       is_training=a     )
       lowercase_		:      Optional[int]          =       ConfigTester(self					,       config_class=a     )




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       pass




    def 			lowerCAmelCase__			(     self					:    Optional[Any]     ):






       '''simple docstring'''






       pass




    def 			lowerCAmelCase__			(     self					:    Union[str, Any]     ):






       '''simple docstring'''






       pass




    def 			lowerCAmelCase__			(     self					:    Optional[Any]     ):






       '''simple docstring'''






       self.config_tester.run_common_tests()




    def 			lowerCAmelCase__			(     self					:    Dict     ):






       '''simple docstring'''






       lowercase_		:      int          =       self.model_tester.prepare_config_and_inputs()
       self.model_tester.create_and_check_decoder_model_past(*a     )





    def 			lowerCAmelCase__			(     self					:    Optional[int]     ):






       '''simple docstring'''






       return












'''simple docstring'''
import os
import unittest

from transformers import RealmTokenizerFast
from transformers.models.bert.tokenization_bert import (
    VOCAB_FILES_NAMES,
    BasicTokenizer,
    WordpieceTokenizer,
    _is_control,
    _is_punctuation,
    _is_whitespace,
)
from transformers.models.realm.tokenization_realm import RealmTokenizer
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin, filter_non_english


@require_tokenizers
class     _UpperCAmelCase  (						snake_case			,   unittest.TestCase     ):
    __lowerCamelCase:				Optional[int]						    =      RealmTokenizer
    __lowerCamelCase:				Optional[Any]						    =      RealmTokenizerFast
    __lowerCamelCase:				Optional[int]						    =      True
    __lowerCamelCase:				str						    =      True
    __lowerCamelCase:				List[str]						    =      filter_non_english




    def 			lowerCAmelCase__			(     self					:    str     ):






       '''simple docstring'''






       super().setUp()

       lowercase_		:      Optional[int]          =       [
           "[UNK]",
           "[CLS]",
           "[SEP]",
           "[PAD]",
           "[MASK]",
           "want",
           "##want",
           "##ed",
           "wa",
           "un",
           "runn",
           "##ing",
           ",",
           "low",
           "lowest",
       ]
       lowercase_		:      List[str]          =       os.path.join(self.tmpdirname					,       VOCAB_FILES_NAMES["vocab_file"]     )
       with open(self.vocab_file					,       "w"					,       encoding="utf-8"     ) as vocab_writer:
          vocab_writer.write("".join([x + "\n" for x in vocab_tokens]     )     )




    def 			lowerCAmelCase__			(     self					:    int					,       a					:    str     ):






       '''simple docstring'''






       lowercase_		:      List[Any]          =       "UNwant\u00E9d,running"
       lowercase_		:      Dict          =       "unwanted, running"
       return input_text, output_text




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       lowercase_		:      str          =       self.tokenizer_class(self.vocab_file     )

       lowercase_		:      Any          =       tokenizer.tokenize("UNwant\u00E9d,running"     )
       self.assertListEqual(a					,       ["un", "##want", "##ed", ",", "runn", "##ing"]     )
       self.assertListEqual(tokenizer.convert_tokens_to_ids(a     )					,       [9, 6, 7, 1_2, 1_0, 1_1]     )




    def 			lowerCAmelCase__			(     self					:    Union[str, Any]     ):






       '''simple docstring'''






       if not self.test_rust_tokenizer:
          return

       lowercase_		:      Optional[Any]          =       self.get_tokenizer()
       lowercase_		:      Any          =       self.get_rust_tokenizer()

       lowercase_		:      Tuple          =       "UNwant\u00E9d,running"

       lowercase_		:      List[str]          =       tokenizer.tokenize(a     )
       lowercase_		:      Union[str, Any]          =       rust_tokenizer.tokenize(a     )
       self.assertListEqual(a					,       a     )

       lowercase_		:      Dict          =       tokenizer.encode(a					,       add_special_tokens=a     )
       lowercase_		:      Dict          =       rust_tokenizer.encode(a					,       add_special_tokens=a     )
       self.assertListEqual(a					,       a     )

       lowercase_		:      List[Any]          =       self.get_rust_tokenizer()
       lowercase_		:      Any          =       tokenizer.encode(a     )
       lowercase_		:      Optional[Any]          =       rust_tokenizer.encode(a     )
       self.assertListEqual(a					,       a     )

       # With lower casing
       lowercase_		:      Union[str, Any]          =       self.get_tokenizer(do_lower_case=a     )
       lowercase_		:      Optional[Any]          =       self.get_rust_tokenizer(do_lower_case=a     )

       lowercase_		:      int          =       "UNwant\u00E9d,running"

       lowercase_		:      List[str]          =       tokenizer.tokenize(a     )
       lowercase_		:      Dict          =       rust_tokenizer.tokenize(a     )
       self.assertListEqual(a					,       a     )

       lowercase_		:      Optional[Any]          =       tokenizer.encode(a					,       add_special_tokens=a     )
       lowercase_		:      Tuple          =       rust_tokenizer.encode(a					,       add_special_tokens=a     )
       self.assertListEqual(a					,       a     )

       lowercase_		:      Dict          =       self.get_rust_tokenizer()
       lowercase_		:      Tuple          =       tokenizer.encode(a     )
       lowercase_		:      int          =       rust_tokenizer.encode(a     )
       self.assertListEqual(a					,       a     )




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       lowercase_		:      Any          =       BasicTokenizer()

       self.assertListEqual(tokenizer.tokenize("ah\u535A\u63A8zz"     )					,       ["ah", "\u535A", "\u63A8", "zz"]     )




    def 			lowerCAmelCase__			(     self					:    Dict     ):






       '''simple docstring'''






       lowercase_		:      str          =       BasicTokenizer(do_lower_case=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHeLLo!how  \n Are yoU?  "     )					,       ["hello", "!", "how", "are", "you", "?"]     )
       self.assertListEqual(tokenizer.tokenize("H\u00E9llo"     )					,       ["hello"]     )




    def 			lowerCAmelCase__			(     self					:    Any     ):






       '''simple docstring'''






       lowercase_		:      Optional[int]          =       BasicTokenizer(do_lower_case=a					,       strip_accents=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHäLLo!how  \n Are yoU?  "     )					,       ["hällo", "!", "how", "are", "you", "?"]     )
       self.assertListEqual(tokenizer.tokenize("H\u00E9llo"     )					,       ["h\u00E9llo"]     )




    def 			lowerCAmelCase__			(     self					:    Any     ):






       '''simple docstring'''






       lowercase_		:      List[Any]          =       BasicTokenizer(do_lower_case=a					,       strip_accents=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHäLLo!how  \n Are yoU?  "     )					,       ["hallo", "!", "how", "are", "you", "?"]     )
       self.assertListEqual(tokenizer.tokenize("H\u00E9llo"     )					,       ["hello"]     )




    def 			lowerCAmelCase__			(     self					:    Union[str, Any]     ):






       '''simple docstring'''






       lowercase_		:      Dict          =       BasicTokenizer(do_lower_case=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHäLLo!how  \n Are yoU?  "     )					,       ["hallo", "!", "how", "are", "you", "?"]     )
       self.assertListEqual(tokenizer.tokenize("H\u00E9llo"     )					,       ["hello"]     )




    def 			lowerCAmelCase__			(     self					:    Union[str, Any]     ):






       '''simple docstring'''






       lowercase_		:      Any          =       BasicTokenizer(do_lower_case=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHeLLo!how  \n Are yoU?  "     )					,       ["HeLLo", "!", "how", "Are", "yoU", "?"]     )




    def 			lowerCAmelCase__			(     self					:    str     ):






       '''simple docstring'''






       lowercase_		:      Union[str, Any]          =       BasicTokenizer(do_lower_case=a					,       strip_accents=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHäLLo!how  \n Are yoU?  "     )					,       ["HäLLo", "!", "how", "Are", "yoU", "?"]     )




    def 			lowerCAmelCase__			(     self					:    Tuple     ):






       '''simple docstring'''






       lowercase_		:      str          =       BasicTokenizer(do_lower_case=a					,       strip_accents=a     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHäLLo!how  \n Are yoU?  "     )					,       ["HaLLo", "!", "how", "Are", "yoU", "?"]     )




    def 			lowerCAmelCase__			(     self					:    Optional[Any]     ):






       '''simple docstring'''






       lowercase_		:      List[Any]          =       BasicTokenizer(do_lower_case=a					,       never_split=["[UNK]"]     )

       self.assertListEqual(
           tokenizer.tokenize(" \tHeLLo!how  \n Are yoU? [UNK]"     )					,       ["HeLLo", "!", "how", "Are", "yoU", "?", "[UNK]"]     )




    def 			lowerCAmelCase__			(     self					:    Tuple     ):






       '''simple docstring'''






       lowercase_		:      int          =       ["[UNK]", "[CLS]", "[SEP]", "want", "##want", "##ed", "wa", "un", "runn", "##ing"]

       lowercase_		:      Tuple          =       {}
       for i, token in enumerate(a     ):
          lowercase_		:      Union[str, Any]          =       i
       lowercase_		:      List[str]          =       WordpieceTokenizer(vocab=a					,       unk_token="[UNK]"     )

       self.assertListEqual(tokenizer.tokenize(""     )					,       []     )

       self.assertListEqual(tokenizer.tokenize("unwanted running"     )					,       ["un", "##want", "##ed", "runn", "##ing"]     )

       self.assertListEqual(tokenizer.tokenize("unwantedX running"     )					,       ["[UNK]", "runn", "##ing"]     )




    def 			lowerCAmelCase__			(     self					:    str     ):






       '''simple docstring'''






       self.assertTrue(_is_whitespace(" "     )     )
       self.assertTrue(_is_whitespace("\t"     )     )
       self.assertTrue(_is_whitespace("\r"     )     )
       self.assertTrue(_is_whitespace("\n"     )     )
       self.assertTrue(_is_whitespace("\u00A0"     )     )

       self.assertFalse(_is_whitespace("A"     )     )
       self.assertFalse(_is_whitespace("-"     )     )




    def 			lowerCAmelCase__			(     self					:    Tuple     ):






       '''simple docstring'''






       self.assertTrue(_is_control("\u0005"     )     )

       self.assertFalse(_is_control("A"     )     )
       self.assertFalse(_is_control(" "     )     )
       self.assertFalse(_is_control("\t"     )     )
       self.assertFalse(_is_control("\r"     )     )




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       self.assertTrue(_is_punctuation("-"     )     )
       self.assertTrue(_is_punctuation("$"     )     )
       self.assertTrue(_is_punctuation("`"     )     )
       self.assertTrue(_is_punctuation("."     )     )

       self.assertFalse(_is_punctuation("A"     )     )
       self.assertFalse(_is_punctuation(" "     )     )




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       lowercase_		:      Dict          =       self.get_tokenizer()

       # Example taken from the issue https://github.com/huggingface/tokenizers/issues/340
       self.assertListEqual([tokenizer.tokenize(a     ) for t in ["Test", "\xad", "test"]]					,       [["[UNK]"], [], ["[UNK]"]]     )

       if self.test_rust_tokenizer:
          lowercase_		:      Dict          =       self.get_rust_tokenizer()
          self.assertListEqual(
              [rust_tokenizer.tokenize(a     ) for t in ["Test", "\xad", "test"]]					,       [["[UNK]"], [], ["[UNK]"]]     )




    @slow
    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       lowercase_		:      Dict          =       self.tokenizer_class.from_pretrained("bert-base-uncased"     )

       lowercase_		:      List[str]          =       tokenizer.encode("sequence builders"					,       add_special_tokens=a     )
       lowercase_		:      Optional[Any]          =       tokenizer.encode("multi-sequence build"					,       add_special_tokens=a     )

       lowercase_		:      Optional[Any]          =       tokenizer.build_inputs_with_special_tokens(a     )
       lowercase_		:      List[str]          =       tokenizer.build_inputs_with_special_tokens(a					,       a     )

       assert encoded_sentence == [1_0_1] + text + [1_0_2]
       assert encoded_pair == [1_0_1] + text + [1_0_2] + text_a + [1_0_2]




    def 			lowerCAmelCase__			(     self					:    List[str]     ):






       '''simple docstring'''






       for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
          with self.subTest(f"""{tokenizer.__class__.__name__} ({pretrained_name})"""     ):
             lowercase_		:      Any          =       self.rust_tokenizer_class.from_pretrained(a					,       **a     )

             lowercase_		:      Dict          =       f"""A, naïve {tokenizer_r.mask_token} AllenNLP sentence."""
             lowercase_		:      Union[str, Any]          =       tokenizer_r.encode_plus(
                 a					,       return_attention_mask=a					,       return_token_type_ids=a					,       return_offsets_mapping=a					,       add_special_tokens=a					,       )

             lowercase_		:      Optional[Any]          =       tokenizer_r.do_lower_case if hasattr(a					,       "do_lower_case"     ) else False
             lowercase_		:      int          =       (
                 [
                     ((0, 0), tokenizer_r.cls_token),
                     ((0, 1), "A"),
                     ((1, 2), ","),
                     ((3, 5), "na"),
                     ((5, 6), "##ï"),
                     ((6, 8), "##ve"),
                     ((9, 1_5), tokenizer_r.mask_token),
                     ((1_6, 2_1), "Allen"),
                     ((2_1, 2_3), "##NL"),
                     ((2_3, 2_4), "##P"),
                     ((2_5, 3_3), "sentence"),
                     ((3_3, 3_4), "."),
                     ((0, 0), tokenizer_r.sep_token),
                 ]
                 if not do_lower_case
                 else [
                     ((0, 0), tokenizer_r.cls_token),
                     ((0, 1), "a"),
                     ((1, 2), ","),
                     ((3, 8), "naive"),
                     ((9, 1_5), tokenizer_r.mask_token),
                     ((1_6, 2_1), "allen"),
                     ((2_1, 2_3), "##nl"),
                     ((2_3, 2_4), "##p"),
                     ((2_5, 3_3), "sentence"),
                     ((3_3, 3_4), "."),
                     ((0, 0), tokenizer_r.sep_token),
                 ]
             )

             self.assertEqual(
                 [e[1] for e in expected_results]					,       tokenizer_r.convert_ids_to_tokens(tokens["input_ids"]     )     )
             self.assertEqual([e[0] for e in expected_results]					,       tokens["offset_mapping"]     )





    @slow
    def 			lowerCAmelCase__			(     self					:    str     ):






       '''simple docstring'''






       for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
          with self.subTest(f"""{tokenizer.__class__.__name__} ({pretrained_name})"""     ):
             lowercase_		:      Optional[Any]          =       self.rust_tokenizer_class.from_pretrained(a					,       **a     )
             lowercase_		:      List[str]          =       self.tokenizer_class.from_pretrained(a					,       **a     )
             lowercase_		:      List[str]          =       [["Hello world!", "Nice to meet you!"], ["The cute cat.", "The adorable dog."]]

             lowercase_		:      int          =       tokenizer_r.batch_encode_candidates(a					,       max_length=1_0					,       return_tensors="np"     )
             lowercase_		:      List[str]          =       tokenizer_p.batch_encode_candidates(a					,       max_length=1_0					,       return_tensors="np"     )

             lowercase_		:      Any          =       (2, 2, 1_0)

             self.assertEqual(encoded_sentence_r["input_ids"].shape					,       a     )
             self.assertEqual(encoded_sentence_r["attention_mask"].shape					,       a     )
             self.assertEqual(encoded_sentence_r["token_type_ids"].shape					,       a     )

             self.assertEqual(encoded_sentence_p["input_ids"].shape					,       a     )
             self.assertEqual(encoded_sentence_p["attention_mask"].shape					,       a     )
             self.assertEqual(encoded_sentence_p["token_type_ids"].shape					,       a     )




