






'''simple docstring'''







from dataclasses import dataclass
from typing import List, Optional, Union

import numpy as np
import PIL
from PIL import Image

from ...utils import BaseOutput, is_torch_available, is_transformers_available




@dataclass
# Copied from diffusers.pipelines.stable_diffusion.__init__.StableDiffusionPipelineOutput with Stable->Alt
class      __SCREAMING_SNAKE_CASE					(__A					):


		"""simple docstring"""


		_a     :   Union[List[PIL.Image.Image], np.ndarray]
		_a     :   Optional[List[bool]]


if is_transformers_available() and is_torch_available():
		from .modeling_roberta_series import RobertaSeriesModelWithTransformation
		from .pipeline_alt_diffusion import AltDiffusionPipeline
		from .pipeline_alt_diffusion_imgaimg import AltDiffusionImgaImgPipeline












'''simple docstring'''







import json
import os
import shutil
import tempfile
import unittest

from transformers.models.wavaveca import WavaVecaCTCTokenizer, WavaVecaFeatureExtractor, WavaVecaProcessor
from transformers.models.wavaveca.tokenization_wavaveca import VOCAB_FILES_NAMES
from transformers.utils import FEATURE_EXTRACTOR_NAME

from .test_feature_extraction_wavaveca import floats_list




class      __SCREAMING_SNAKE_CASE					(unittest.TestCase					):


  """simple docstring"""


  def       _a					( self  ):






         """simple docstring"""


         a_	    = '<pad> <s> </s> <unk> | E T A O N I H S R D L U M W C F G Y P B V K \' X J Q Z'.split(' '  )
         a_	    = dict(zip(UpperCamelCase__      ,  range(len(UpperCamelCase__  )  )  )  )

         a_	    = {
             'pad_token': '<pad>',
             'unk_token': '<unk>',
             'bos_token': '<s>',
             'eos_token': '</s>',
         }
         a_	    = {
             'feature_size': 1,
             'padding_value': 0.0,
             'sampling_rate': 16_000,
             'return_attention_mask': False,
             'do_normalize': True,
         }

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

         with open(self.feature_extraction_file      ,  'w'      ,  encoding='utf-8'  ) as fp:
                fp.write(json.dumps(UpperCamelCase__  ) + '\n'  )







  def       _a					( self      ,  **UpperCamelCase__  ):






         """simple docstring"""


         a_	    = self.add_kwargs_tokens_map.copy()
         kwargs.update(UpperCamelCase__  )
         return WavaVecaCTCTokenizer.from_pretrained(self.tmpdirname      ,  **UpperCamelCase__  )







  def       _a					( self      ,  **UpperCamelCase__  ):






         """simple docstring"""


         return WavaVecaFeatureExtractor.from_pretrained(self.tmpdirname      ,  **UpperCamelCase__  )







  def       _a					( self  ):






         """simple docstring"""


         shutil.rmtree(self.tmpdirname  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = self.get_tokenizer()
         a_	    = self.get_feature_extractor()

         a_	    = WavaVecaProcessor(tokenizer=UpperCamelCase__      ,  feature_extractor=UpperCamelCase__  )

         processor.save_pretrained(self.tmpdirname  )
         a_	    = WavaVecaProcessor.from_pretrained(self.tmpdirname  )

         self.assertEqual(processor.tokenizer.get_vocab()      ,  tokenizer.get_vocab()  )
         self.assertIsInstance(processor.tokenizer      ,  UpperCamelCase__  )

         self.assertEqual(processor.feature_extractor.to_json_string()      ,  feature_extractor.to_json_string()  )
         self.assertIsInstance(processor.feature_extractor      ,  UpperCamelCase__  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = WavaVecaProcessor(tokenizer=self.get_tokenizer()      ,  feature_extractor=self.get_feature_extractor()  )
         processor.save_pretrained(self.tmpdirname  )

         a_	    = self.get_tokenizer(bos_token='(BOS)'      ,  eos_token='(EOS)'  )
         a_	    = self.get_feature_extractor(do_normalize=UpperCamelCase__      ,  padding_value=1.0  )

         a_	    = WavaVecaProcessor.from_pretrained(
             self.tmpdirname      ,  bos_token='(BOS)'      ,  eos_token='(EOS)'      ,  do_normalize=UpperCamelCase__      ,  padding_value=1.0  )

         self.assertEqual(processor.tokenizer.get_vocab()      ,  tokenizer_add_kwargs.get_vocab()  )
         self.assertIsInstance(processor.tokenizer      ,  UpperCamelCase__  )

         self.assertEqual(processor.feature_extractor.to_json_string()      ,  feature_extractor_add_kwargs.to_json_string()  )
         self.assertIsInstance(processor.feature_extractor      ,  UpperCamelCase__  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = self.get_feature_extractor()
         a_	    = self.get_tokenizer()

         a_	    = WavaVecaProcessor(tokenizer=UpperCamelCase__      ,  feature_extractor=UpperCamelCase__  )

         a_	    = floats_list((3, 1_000)  )

         a_	    = feature_extractor(UpperCamelCase__      ,  return_tensors='np'  )
         a_	    = processor(UpperCamelCase__      ,  return_tensors='np'  )

         for key in input_feat_extract.keys():
                self.assertAlmostEqual(input_feat_extract[key].sum()      ,  input_processor[key].sum()      ,  delta=1e-2  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = self.get_feature_extractor()
         a_	    = self.get_tokenizer()

         a_	    = WavaVecaProcessor(tokenizer=UpperCamelCase__      ,  feature_extractor=UpperCamelCase__  )

         a_	    = 'This is a test string'

         a_	    = processor(text=UpperCamelCase__  )

         a_	    = tokenizer(UpperCamelCase__  )

         for key in encoded_tok.keys():
                self.assertListEqual(encoded_tok[key]      ,  encoded_processor[key]  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = self.get_feature_extractor()
         a_	    = self.get_tokenizer()

         a_	    = WavaVecaProcessor(tokenizer=UpperCamelCase__      ,  feature_extractor=UpperCamelCase__  )

         a_	    = [[1, 4, 5, 8, 1, 0, 8], [3, 4, 3, 1, 1, 8, 9]]

         a_	    = processor.batch_decode(UpperCamelCase__  )
         a_	    = tokenizer.batch_decode(UpperCamelCase__  )

         self.assertListEqual(UpperCamelCase__      ,  UpperCamelCase__  )







  def       _a					( self  ):






         """simple docstring"""


         a_	    = self.get_feature_extractor()
         a_	    = self.get_tokenizer()

         a_	    = WavaVecaProcessor(tokenizer=UpperCamelCase__      ,  feature_extractor=UpperCamelCase__  )

         self.assertListEqual(
             processor.model_input_names      ,  feature_extractor.model_input_names      ,  msg='`processor` and `feature_extractor` model input names do not match'      ,  )




