


"""simple docstring"""





import unittest

from transformers.models.xlm_prophetnet.tokenization_xlm_prophetnet import SPIECE_UNDERLINE, XLMProphetNetTokenizer
from transformers.testing_utils import get_tests_dir, require_sentencepiece, slow
from transformers.utils import cached_property

from ...test_tokenization_common import TokenizerTesterMixin


snake_case__						:   Union[str, Any]			 = get_tests_dir('''fixtures/test_sentencepiece.model''')



@require_sentencepiece
class 						snake_case_(				a__							,    unittest.TestCase	):
     __UpperCamelCase     =    XLMProphetNetTokenizer
     __UpperCamelCase     =    False
     __UpperCamelCase     =    True






     def    lowerCamelCase__   (		self     :    List[str] ):
           super().setUp()

           # We have a SentencePiece fixture for testing
           lowerCAmelCase    :		str				=      XLMProphetNetTokenizer(UpperCamelCase_    , keep_accents=UpperCamelCase_ )
           tokenizer.save_pretrained(self.tmpdirname )






     def    lowerCamelCase__   (		self     :    List[Any] ):
           lowerCAmelCase    :		int				=      '''[PAD]'''
           lowerCAmelCase    :		Optional[int]				=      0

           self.assertEqual(self.get_tokenizer()._convert_token_to_id(UpperCamelCase_ )    , UpperCamelCase_ )
           self.assertEqual(self.get_tokenizer()._convert_id_to_token(UpperCamelCase_ )    , UpperCamelCase_ )






     def    lowerCamelCase__   (		self     :    List[str] ):
           lowerCAmelCase    :		Tuple				=      list(self.get_tokenizer().get_vocab().keys() )

           self.assertEqual(vocab_keys[0]    , '''[PAD]''' )
           self.assertEqual(vocab_keys[1]    , '''[CLS]''' )
           self.assertEqual(vocab_keys[-1]    , '''j''' )
           self.assertEqual(len(UpperCamelCase_ )    , 1_0_1_2 )






     def    lowerCamelCase__   (		self     :    Tuple ):
           self.assertEqual(self.get_tokenizer().vocab_size    , 1_0_1_2 )






     def    lowerCamelCase__   (		self     :    List[str] ):
           lowerCAmelCase    :		List[str]				=      XLMProphetNetTokenizer(UpperCamelCase_    , keep_accents=UpperCamelCase_ )

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

           self.assertListEqual(
               tokenizer.convert_tokens_to_ids(UpperCamelCase_ )    , [value + tokenizer.fairseq_offset for value in [2_8_5, 4_6, 1_0, 1_7_0, 3_8_2]]    , )

           lowerCAmelCase    :		Optional[Any]				=      tokenizer.tokenize('''I was born in 92000, and this is falsé.''' )
           self.assertListEqual(
               UpperCamelCase_    , [
                   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''',
                   '''é''',
                   '''.''',
               ]    , )
           lowerCAmelCase    :		Dict				=      tokenizer.convert_tokens_to_ids(UpperCamelCase_ )
           self.assertListEqual(
               UpperCamelCase_    , [
                   value + tokenizer.fairseq_offset
                   for value in [8, 2_1, 8_4, 5_5, 2_4, 1_9, 7, -9, 6_0_2, 3_4_7, 3_4_7, 3_4_7, 3, 1_2, 6_6, 4_6, 7_2, 8_0, 6, -9, 4]
               ]    , )

           lowerCAmelCase    :		int				=      tokenizer.convert_ids_to_tokens(UpperCamelCase_ )
           self.assertListEqual(
               UpperCamelCase_    , [
                   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]''',
                   '''.''',
               ]    , )






     @cached_property
     def    lowerCamelCase__   (		self     :    Union[str, Any] ):
           return XLMProphetNetTokenizer.from_pretrained('''microsoft/xprophetnet-large-wiki100-cased''' )






     @slow
     def    lowerCamelCase__   (		self     :    Dict ):
           lowerCAmelCase    :		List[Any]				=      '''Hello World!'''
           lowerCAmelCase    :		Union[str, Any]				=      [3_5_3_8_9, 6_6_7_2, 4_9, 2]
           self.assertListEqual(UpperCamelCase_    , self.big_tokenizer.encode(UpperCamelCase_ ) )






     @slow
     def    lowerCamelCase__   (		self     :    Union[str, Any] ):
           # fmt: off
           lowerCAmelCase    :		str				=      {'''input_ids''': [[1_1_0_7_3, 8_2_7_8_3, 1_8, 2_6, 8_2_7_8_3, 5_4_9, 5_1_5_4_0, 2_4_8, 1_7_2_0_9, 1_3_0_1, 2_1_7, 2_0, 2_1_5_1_8_6, 1_3_2_5, 1_4_7, 1_7_2_0_9, 1_3_0_1, 2_1_7, 2_0, 5_6_3_7_0, 5_3, 1_2_2_0_2_0, 2_0, 1_6_4_7_7, 2_7, 8_7_3_5_5, 4_5_4_8, 2_0, 4_7_2_8, 7_8_3_9_2, 1_7, 1_5_9_9_6_9, 1_8, 2_6, 2_4_4_9_1, 6_2_9, 1_5, 5_3_8, 2_2_7_0_4, 5_4_3_9, 1_5, 2_7_8_8, 2_4_4_9_1, 9_8_8_5, 1_5, 4_3_5_3_4, 6_0_5, 1_5, 8_1_4, 1_8_4_0_3, 3_3_2_0_0, 2_9, 1_5, 4_3_5_3_4, 2_4_4_5_8, 1_2_4_1_0, 1_1_1, 2_4_9_6_6, 8_3_6_6_9, 9_6_3_7, 1_4_4_0_6_8, 2_6, 8_5_0, 2_2_3_4_6, 2_7, 1_4_7, 2_4_9_6_6, 8_3_6_6_9, 8_3_4_9_0, 2_6, 3_9_1_1_3, 7_3_5, 2_7, 6_8_9, 6_5_6, 2_8_0_0, 1_3_3_9, 4_6_0_0, 5_3, 1_2_2_0_2_0, 1_1_5_7_8_5, 3_4, 8_1_6, 1_3_3_9, 4_6_8_8_7, 1_8, 1_4_7, 5_3_9_0_5, 1_9_5_1, 4_2_2_3_8, 4_1_1_7_0, 1_7_7_3_2, 8_3_4, 4_3_6, 1_5, 2_7_5_2_3, 9_8_7_3_3, 2_1_7, 1_4_7, 5_5_4_2, 4_9_8_1, 9_3_0, 1_7_3_4_7, 1_6, 2], [2_0_0_9_1, 6_2_9, 9_4, 8_2_7_8_6, 5_8, 4_9_0, 2_0, 1_5_2_8, 8_4, 5_3_9_0_5, 3_4_4, 8_0_5_9_2, 1_1_0_1_2_8, 1_8_8_2_2, 5_2_6_7, 1_3_0_6, 6_2, 1_5_2_5_3_7, 3_0_8, 7_9_9_7, 4_0_1, 1_2_4_4_2_7, 5_4_9, 3_5_4_4_2, 2_2_5, 1_0_9, 1_5_0_5_5, 2_5_7_4_8, 1_4_7, 7_1_1_9, 4_3_7_1_2, 3_4, 7_6_7, 1_3_5_3_6_6, 1_8, 1_6, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [5_9_2, 6_3_7_8_4, 1_1_9_4_6_6, 1_7, 1_4_7_8_0_8, 8_8_2_1_4, 1_8, 6_5_6, 8_1, 3_2, 3_2_9_6, 1_0_2_8_0, 1_6, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], '''attention_mask''': [[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]}  # noqa: E501
           # fmt: on

           self.tokenizer_integration_test_util(
               expected_encoding=UpperCamelCase_    , model_name='''microsoft/xprophetnet-large-wiki100-cased'''    , revision='''1acad1643ddd54a44df6a1b797ada8373685d90e'''    , )






"""simple docstring"""





import math


def    _snake_case  (  ):
      lowerCAmelCase    :		Union[str, Any]				=      input('''Enter message: '''	)
      lowerCAmelCase    :		Optional[int]				=      int(input(f'''Enter key [2-{len(_snake_case	) - 1}]: '''	)	)
      lowerCAmelCase    :		str				=      input('''Encryption/Decryption [e/d]: '''	)

      if mode.lower().startswith('''e'''	):
            lowerCAmelCase    :		Any				=      encrypt_message(_snake_case       ,      _snake_case	)
      elif mode.lower().startswith('''d'''	):
            lowerCAmelCase    :		Union[str, Any]				=      decrypt_message(_snake_case       ,      _snake_case	)

      # Append pipe symbol (vertical bar) to identify spaces at the end.
      print(f'''Output:\n{text + "|"}'''	)


def    _snake_case  (  _snake_case :	int       ,      _snake_case :	str	):
      lowerCAmelCase    :		Optional[Any]				=      [''''''] * key
      for col in range(_snake_case	):
            lowerCAmelCase    :		Optional[Any]				=      col
            while pointer < len(_snake_case	):
                  cipher_text[col] += message[pointer]
                  pointer += key
      return "".join(_snake_case	)


def    _snake_case  (  _snake_case :	int       ,      _snake_case :	str	):
      lowerCAmelCase    :		Union[str, Any]				=      math.ceil(len(_snake_case	) / key	)
      lowerCAmelCase    :		str				=      key
      lowerCAmelCase    :		Any				=      (num_cols * num_rows) - len(_snake_case	)
      lowerCAmelCase    :		Dict				=      [''''''] * num_cols
      lowerCAmelCase    :		int				=      0
      lowerCAmelCase    :		int				=      0

      for symbol in message:
            plain_text[col] += symbol
            col += 1

            if (
                (col == num_cols)
                or (col == num_cols - 1)
                and (row >= num_rows - num_shaded_boxes)
            ):
                  lowerCAmelCase    :		int				=      0
                  row += 1

      return "".join(_snake_case	)


if __name__ == "__main__":
 import doctest

 doctest.testmod()
 main()


