





'''simple docstring'''







import json
import os
import re
import unittest

from transformers import CodeGenTokenizer, CodeGenTokenizerFast
from transformers.models.codegen.tokenization_codegen import VOCAB_FILES_NAMES
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin
@require_tokenizers
class 			_lowercase						(       __a				,		unittest.TestCase					):
     _UpperCAmelCase										=					CodeGenTokenizer
     _UpperCAmelCase										=					CodeGenTokenizerFast
     _UpperCAmelCase										=					True
     _UpperCAmelCase										=					{'''add_prefix_space''': True}
     _UpperCAmelCase										=					False







     def     UpperCamelCase			( self    ) ->	Tuple:
      super().setUp()

      # Adapted from Sennrich et al. 2015 and https://github.com/rsennrich/subword-nmt
      snake_case				=			[
          '''l''',
          '''o''',
          '''w''',
          '''e''',
          '''r''',
          '''s''',
          '''t''',
          '''i''',
          '''d''',
          '''n''',
          '''\u0120''',
          '''\u0120l''',
          '''\u0120n''',
          '''\u0120lo''',
          '''\u0120low''',
          '''er''',
          '''\u0120lowest''',
          '''\u0120newer''',
          '''\u0120wider''',
          '''<unk>''',
          '''<|endoftext|>''',
      ]
      snake_case				=			dict(zip(A__    ,   range(len(A__    )    )    )    )
      snake_case				=			['''#version: 0.2''', '''\u0120 l''', '''\u0120l o''', '''\u0120lo w''', '''e r''', '''''']
      snake_case				=			{'''unk_token''': '''<unk>'''}

      snake_case				=			os.path.join(self.tmpdirname    ,   VOCAB_FILES_NAMES['''vocab_file''']    )
      snake_case				=			os.path.join(self.tmpdirname    ,   VOCAB_FILES_NAMES['''merges_file''']    )
      with open(self.vocab_file    ,   '''w'''    ,   encoding='''utf-8'''    ) as fp:
       fp.write(json.dumps(A__    ) + '''\n'''    )
      with open(self.merges_file    ,   '''w'''    ,   encoding='''utf-8'''    ) as fp:
       fp.write('''\n'''.join(A__    )    )







     def     UpperCamelCase			( self    ,   **A__    ) ->	Union[str, Any]:
      kwargs.update(self.special_tokens_map    )
      return CodeGenTokenizer.from_pretrained(self.tmpdirname    ,   **A__    )







     def     UpperCamelCase			( self    ,   **A__    ) ->	Union[str, Any]:
      kwargs.update(self.special_tokens_map    )
      return CodeGenTokenizerFast.from_pretrained(self.tmpdirname    ,   **A__    )







     def     UpperCamelCase			( self    ,   A__    ) ->	Tuple:
      snake_case				=			'''lower newer'''
      snake_case				=			'''lower newer'''
      return input_text, output_text







     def     UpperCamelCase			( self    ) ->	List[Any]:
      snake_case				=			CodeGenTokenizer(self.vocab_file    ,   self.merges_file    ,   **self.special_tokens_map    )
      snake_case				=			'''lower newer'''
      snake_case				=			['''\u0120low''', '''er''', '''\u0120''', '''n''', '''e''', '''w''', '''er''']
      snake_case				=			tokenizer.tokenize(A__    ,   add_prefix_space=A__    )
      self.assertListEqual(A__    ,   A__    )

      snake_case				=			tokens + [tokenizer.unk_token]
      snake_case				=			[14, 15, 10, 9, 3, 2, 15, 19]
      self.assertListEqual(tokenizer.convert_tokens_to_ids(A__    )    ,   A__    )







     def     UpperCamelCase			( self    ) ->	Optional[int]:
      if not self.test_rust_tokenizer:
       return

      snake_case				=			self.get_tokenizer()
      snake_case				=			self.get_rust_tokenizer(add_prefix_space=A__    )

      snake_case				=			'''lower newer'''

      # Testing tokenization
      snake_case				=			tokenizer.tokenize(A__    ,   add_prefix_space=A__    )
      snake_case				=			rust_tokenizer.tokenize(A__    )
      self.assertListEqual(A__    ,   A__    )

      # Testing conversion to ids without special tokens
      snake_case				=			tokenizer.encode(A__    ,   add_special_tokens=A__    ,   add_prefix_space=A__    )
      snake_case				=			rust_tokenizer.encode(A__    ,   add_special_tokens=A__    )
      self.assertListEqual(A__    ,   A__    )

      # Testing conversion to ids with special tokens
      snake_case				=			self.get_rust_tokenizer(add_prefix_space=A__    )
      snake_case				=			tokenizer.encode(A__    ,   add_prefix_space=A__    )
      snake_case				=			rust_tokenizer.encode(A__    )
      self.assertListEqual(A__    ,   A__    )

      # Testing the unknown token
      snake_case				=			tokens + [rust_tokenizer.unk_token]
      snake_case				=			[14, 15, 10, 9, 3, 2, 15, 19]
      self.assertListEqual(rust_tokenizer.convert_tokens_to_ids(A__    )    ,   A__    )







     def     UpperCamelCase			( self    ,   *A__    ,   **A__    ) ->	List[str]:
      # It's very difficult to mix/test pretokenization with byte-level
      # And get both CodeGen and Roberta to work at the same time (mostly an issue of adding a space before the string)
      pass







     def     UpperCamelCase			( self    ,   A__=15    ) ->	Tuple:
      for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
       with self.subTest(F"""{tokenizer.__class__.__name__} ({pretrained_name})"""    ):
        snake_case				=			self.rust_tokenizer_class.from_pretrained(A__    ,   **A__    )

        # Simple input
        snake_case				=			'''This is a simple input'''
        snake_case				=			['''This is a simple input 1''', '''This is a simple input 2''']
        snake_case				=			('''This is a simple input''', '''This is a pair''')
        snake_case				=			[
            ('''This is a simple input 1''', '''This is a simple input 2'''),
            ('''This is a simple pair 1''', '''This is a simple pair 2'''),
        ]

        # Simple input tests
        self.assertRaises(A__    ,   tokenizer_r.encode    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    )

        # Simple input
        self.assertRaises(A__    ,   tokenizer_r.encode_plus    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    )

        # Simple input
        self.assertRaises(
            A__    ,   tokenizer_r.batch_encode_plus    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    ,   )

        # Pair input
        self.assertRaises(A__    ,   tokenizer_r.encode    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    )

        # Pair input
        self.assertRaises(A__    ,   tokenizer_r.encode_plus    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    )

        # Pair input
        self.assertRaises(
            A__    ,   tokenizer_r.batch_encode_plus    ,   A__    ,   max_length=A__    ,   padding='''max_length'''    ,   )







     def     UpperCamelCase			( self    ) ->	Tuple:
      snake_case				=			CodeGenTokenizer.from_pretrained(self.tmpdirname    ,   pad_token='''<pad>'''    )

      # Simple input
      snake_case				=			'''This is a simple input'''
      snake_case				=			['''This is a simple input looooooooong''', '''This is a simple input''']
      snake_case				=			('''This is a simple input''', '''This is a pair''')
      snake_case				=			[
          ('''This is a simple input loooooong''', '''This is a simple input'''),
          ('''This is a simple pair loooooong''', '''This is a simple pair'''),
      ]

      snake_case				=			tokenizer.pad_token_id

      snake_case				=			tokenizer(A__    ,   padding='''max_length'''    ,   max_length=30    ,   return_tensors='''np'''    )
      snake_case				=			tokenizer(A__    ,   padding=A__    ,   truncate=A__    ,   return_tensors='''np'''    )
      snake_case				=			tokenizer(*A__    ,   padding='''max_length'''    ,   max_length=60    ,   return_tensors='''np'''    )
      snake_case				=			tokenizer(A__    ,   padding=A__    ,   truncate=A__    ,   return_tensors='''np'''    )

      # s
      # test single string max_length padding
      self.assertEqual(out_s['''input_ids'''].shape[-1]    ,   30    )
      self.assertTrue(pad_token_id in out_s['''input_ids''']    )
      self.assertTrue(0 in out_s['''attention_mask''']    )

      # s2
      # test automatic padding
      self.assertEqual(out_sa['''input_ids'''].shape[-1]    ,   33    )
      # long slice doesn't have padding
      self.assertFalse(pad_token_id in out_sa['''input_ids'''][0]    )
      self.assertFalse(0 in out_sa['''attention_mask'''][0]    )
      # short slice does have padding
      self.assertTrue(pad_token_id in out_sa['''input_ids'''][1]    )
      self.assertTrue(0 in out_sa['''attention_mask'''][1]    )

      # p
      # test single pair max_length padding
      self.assertEqual(out_p['''input_ids'''].shape[-1]    ,   60    )
      self.assertTrue(pad_token_id in out_p['''input_ids''']    )
      self.assertTrue(0 in out_p['''attention_mask''']    )

      # p2
      # test automatic padding pair
      self.assertEqual(out_pa['''input_ids'''].shape[-1]    ,   52    )
      # long slice pair doesn't have padding
      self.assertFalse(pad_token_id in out_pa['''input_ids'''][0]    )
      self.assertFalse(0 in out_pa['''attention_mask'''][0]    )
      # short slice pair does have padding
      self.assertTrue(pad_token_id in out_pa['''input_ids'''][1]    )
      self.assertTrue(0 in out_pa['''attention_mask'''][1]    )







     def     UpperCamelCase			( self    ) ->	str:
      snake_case				=			'''$$$'''
      snake_case				=			CodeGenTokenizer.from_pretrained(self.tmpdirname    ,   bos_token=A__    ,   add_bos_token=A__    )

      snake_case				=			'''This is a simple input'''
      snake_case				=			['''This is a simple input 1''', '''This is a simple input 2''']

      snake_case				=			tokenizer.bos_token_id

      snake_case				=			tokenizer(A__    )
      snake_case				=			tokenizer(A__    )

      self.assertEqual(out_s.input_ids[0]    ,   A__    )
      self.assertTrue(all(o[0] == bos_token_id for o in out_sa.input_ids    )    )

      snake_case				=			tokenizer.decode(out_s.input_ids    )
      snake_case				=			tokenizer.batch_decode(out_sa.input_ids    )

      self.assertEqual(decode_s.split()[0]    ,   A__    )
      self.assertTrue(all(d.split()[0] == bos_token for d in decode_sa    )    )







     @slow
     def     UpperCamelCase			( self    ) ->	Any:
      snake_case				=			CodeGenTokenizer.from_pretrained('''Salesforce/codegen-350M-mono'''    )

      snake_case				=			'''\nif len_a > len_b:\n    result = a\nelse:\n    result = b\n\n\n\n#'''
      snake_case				=			'''\nif len_a > len_b:      result = a\nelse:      result = b'''

      snake_case				=			tokenizer.encode(A__    )
      snake_case				=			['''^#''', re.escape('''<|endoftext|>'''    ), '''^\'\'\'''', '''^"""''', '''\n\n\n''']
      snake_case				=			tokenizer.decode(A__    ,   truncate_before_pattern=A__    )
      self.assertEqual(A__    ,   A__    )







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








'''simple docstring'''







import os
from dataclasses import dataclass, field
from io import BytesIO
from typing import TYPE_CHECKING, Any, ClassVar, Dict, Optional, Union

import numpy as np
import pyarrow as pa

from .. import config
from ..download.streaming_download_manager import xopen, xsplitext
from ..table import array_cast
from ..utils.py_utils import no_op_if_value_is_null, string_to_dict


if TYPE_CHECKING:
   from .features import FeatureType


_lowercase						,				_lowercase						,				_lowercase       						=			False, False, False
@dataclass
class 			_lowercase						:
     _UpperCAmelCase										=					None
     _UpperCAmelCase										=					True
     _UpperCAmelCase										=					True
     _UpperCAmelCase										=					None
     # Automatically constructed
     _UpperCAmelCase										=					"dict"
     _UpperCAmelCase										=					pa.struct({'''bytes''': pa.binary(), '''path''': pa.string()}					)
     _UpperCAmelCase										=					field(default='''Audio'''				,		init=__a				,		repr=__a					)







     def __call__( self    ) ->	Optional[Any]:
      return self.pa_type







     def     UpperCamelCase			( self    ,   A__    ) ->	dict:
      try:
       import soundfile as sf  # soundfile is a dependency of librosa, needed to decode audio files.
      except ImportError as err:
       raise ImportError('''To support encoding audio data, please install \'soundfile\'.'''    ) from err
      if isinstance(A__    ,   A__    ):
       return {"bytes": None, "path": value}
      elif isinstance(A__    ,   A__    ):
       return {"bytes": value, "path": None}
      elif "array" in value:
       # convert the audio array to wav bytes
       snake_case				=			BytesIO()
       sf.write(A__    ,   value['''array''']    ,   value['''sampling_rate''']    ,   format='''wav'''    )
       return {"bytes": buffer.getvalue(), "path": None}
      elif value.get('''path'''    ) is not None and os.path.isfile(value['''path''']    ):
       # we set "bytes": None to not duplicate the data if they're already available locally
       if value["path"].endswith('''pcm'''    ):
        # "PCM" only has raw audio bytes
        if value.get('''sampling_rate'''    ) is None:
         # At least, If you want to convert "PCM-byte" to "WAV-byte", you have to know sampling rate
         raise KeyError('''To use PCM files, please specify a \'sampling_rate\' in Audio object'''    )
        if value.get('''bytes'''    ):
         # If we already had PCM-byte, we don`t have to make "read file, make bytes" (just use it!)
         snake_case				=			np.frombuffer(value['''bytes''']    ,   dtype=np.intaa    ).astype(np.floataa    ) / 3_27_67
        else:
         snake_case				=			np.memmap(value['''path''']    ,   dtype='''h'''    ,   mode='''r'''    ).astype(np.floataa    ) / 3_27_67

        snake_case				=			BytesIO(bytes()    )
        sf.write(A__    ,   A__    ,   value['''sampling_rate''']    ,   format='''wav'''    )
        return {"bytes": buffer.getvalue(), "path": None}
       else:
        return {"bytes": None, "path": value.get('''path'''    )}
      elif value.get('''bytes'''    ) is not None or value.get('''path'''    ) is not None:
       # store the audio bytes, and path is used to infer the audio format using the file extension
       return {"bytes": value.get('''bytes'''    ), "path": value.get('''path'''    )}
      else:
       raise ValueError(
           F"""An audio sample should have one of 'path' or 'bytes' but they are missing or None in {value}."""    )







     def     UpperCamelCase			( self    ,   A__    ,   A__ = None    ) ->	dict:
      if not self.decode:
       raise RuntimeError('''Decoding is disabled for this feature. Please use Audio(decode=True) instead.'''    )

      snake_case				,		snake_case				=			(value['''path'''], BytesIO(value['''bytes''']    )) if value['''bytes'''] is not None else (value['''path'''], None)
      if path is None and file is None:
       raise ValueError(F"""An audio sample should have one of 'path' or 'bytes' but both are None in {value}."""    )

      try:
       import librosa
       import soundfile as sf
      except ImportError as err:
       raise ImportError('''To support decoding audio files, please install \'librosa\' and \'soundfile\'.'''    ) from err

      snake_case				=			xsplitext(A__    )[1][1:].lower() if path is not None else None
      if not config.IS_OPUS_SUPPORTED and audio_format == "opus":
       raise RuntimeError(
           '''Decoding \'opus\' files requires system library \'libsndfile\'>=1.0.31, '''
           '''You can try to update `soundfile` python library: `pip install "soundfile>=0.12.1"`. '''    )
      elif not config.IS_MP3_SUPPORTED and audio_format == "mp3":
       raise RuntimeError(
           '''Decoding \'mp3\' files requires system library \'libsndfile\'>=1.1.0, '''
           '''You can try to update `soundfile` python library: `pip install "soundfile>=0.12.1"`. '''    )

      if file is None:
       snake_case				=			token_per_repo_id or {}
       snake_case				=			path.split('''::'''    )[-1]
       try:
        snake_case				=			string_to_dict(A__    ,   config.HUB_DATASETS_URL    )['''repo_id''']
        snake_case				=			token_per_repo_id[repo_id]
       except (ValueError, KeyError):
        snake_case				=			None

       with xopen(A__    ,   '''rb'''    ,   use_auth_token=A__    ) as f:
        snake_case				,		snake_case				=			sf.read(A__    )

      else:
       snake_case				,		snake_case				=			sf.read(A__    )

      snake_case				=			array.T
      if self.mono:
       snake_case				=			librosa.to_mono(A__    )
      if self.sampling_rate and self.sampling_rate != sampling_rate:
       snake_case				=			librosa.resample(A__    ,   orig_sr=A__    ,   target_sr=self.sampling_rate    )
       snake_case				=			self.sampling_rate

      return {"path": path, "array": array, "sampling_rate": sampling_rate}







     def     UpperCamelCase			( self    ) ->	Union["FeatureType", Dict[str, "FeatureType"]]:
      from .features import Value

      if self.decode:
       raise ValueError('''Cannot flatten a decoded Audio feature.'''    )
      return {
          "bytes": Value('''binary'''    ),
          "path": Value('''string'''    ),
      }







     def     UpperCamelCase			( self    ,   A__    ) ->	pa.StructArray:
      if pa.types.is_string(storage.type    ):
       snake_case				=			pa.array([None] * len(A__    )    ,   type=pa.binary()    )
       snake_case				=			pa.StructArray.from_arrays([bytes_array, storage]    ,   ['''bytes''', '''path''']    ,   mask=storage.is_null()    )
      elif pa.types.is_binary(storage.type    ):
       snake_case				=			pa.array([None] * len(A__    )    ,   type=pa.string()    )
       snake_case				=			pa.StructArray.from_arrays([storage, path_array]    ,   ['''bytes''', '''path''']    ,   mask=storage.is_null()    )
      elif pa.types.is_struct(storage.type    ) and storage.type.get_all_field_indices('''array'''    ):
       snake_case				=			pa.array([Audio().encode_example(A__    ) if x is not None else None for x in storage.to_pylist()]    )
      elif pa.types.is_struct(storage.type    ):
       if storage.type.get_field_index('''bytes'''    ) >= 0:
        snake_case				=			storage.field('''bytes'''    )
       else:
        snake_case				=			pa.array([None] * len(A__    )    ,   type=pa.binary()    )
       if storage.type.get_field_index('''path'''    ) >= 0:
        snake_case				=			storage.field('''path'''    )
       else:
        snake_case				=			pa.array([None] * len(A__    )    ,   type=pa.string()    )
       snake_case				=			pa.StructArray.from_arrays([bytes_array, path_array]    ,   ['''bytes''', '''path''']    ,   mask=storage.is_null()    )
      return array_cast(A__    ,   self.pa_type    )







     def     UpperCamelCase			( self    ,   A__    ) ->	pa.StructArray:

      @no_op_if_value_is_null
      def path_to_bytes(A__    ):
       with xopen(A__    ,   '''rb'''    ) as f:
        snake_case				=			f.read()
       return bytes_

      snake_case				=			pa.array(
          [
              (path_to_bytes(x['''path''']    ) if x['''bytes'''] is None else x['''bytes''']) if x is not None else None
              for x in storage.to_pylist()
          ]    ,   type=pa.binary()    ,   )
      snake_case				=			pa.array(
          [os.path.basename(A__    ) if path is not None else None for path in storage.field('''path'''    ).to_pylist()]    ,   type=pa.string()    ,   )
      snake_case				=			pa.StructArray.from_arrays([bytes_array, path_array]    ,   ['''bytes''', '''path''']    ,   mask=bytes_array.is_null()    )
      return array_cast(A__    ,   self.pa_type    )

