






from typing import List, Optional, Union

import numpy as np
import torch
import torchaudio.compliance.kaldi as ta_kaldi

from ...feature_extraction_sequence_utils import SequenceFeatureExtractor
from ...feature_extraction_utils import BatchFeature
from ...utils import PaddingStrategy, TensorType, logging


lowerCAmelCase_           = logging.get_logger(__name__)
class      UpperCamelCase (				snake_case__ ):







		"""simple docstring"""



		snake_case    =   ["input_features", "attention_mask"]
		def __init__(  self		:       str						,_SCREAMING_SNAKE_CASE		:       Optional[Any]=8_0						,_SCREAMING_SNAKE_CASE		:       Union[str, Any]=1_6_0_0_0						,_SCREAMING_SNAKE_CASE		:       Dict=8_0						,_SCREAMING_SNAKE_CASE		:       Optional[int]=0.0						,_SCREAMING_SNAKE_CASE		:       int=True						,_SCREAMING_SNAKE_CASE		:       List[Any]=True						,_SCREAMING_SNAKE_CASE		:       str=True						,**_SCREAMING_SNAKE_CASE		:       Optional[Any]						,)    ->   Dict:


						'''simple docstring'''
						super().__init__(feature_size=_SCREAMING_SNAKE_CASE						,sampling_rate=_SCREAMING_SNAKE_CASE						,padding_value=_SCREAMING_SNAKE_CASE						,**_SCREAMING_SNAKE_CASE		)
						A	 =      num_mel_bins
						A	 =      do_ceptral_normalize
						A	 =      normalize_means
						A	 =      normalize_vars
						A	 =      True
		def       A(  self		:       Optional[int]						,_SCREAMING_SNAKE_CASE		:       np.ndarray						,)    ->   np.ndarray:


						'''simple docstring'''
						A	 =      waveform * (2**1_5)  # Kaldi compliance: 16-bit signed integers
						A	 =      torch.from_numpy(_SCREAMING_SNAKE_CASE		).unsqueeze(0		)
						A	 =      ta_kaldi.fbank(_SCREAMING_SNAKE_CASE						,num_mel_bins=self.num_mel_bins						,sample_frequency=self.sampling_rate		)
						return features.numpy()
		@staticmethod
		def       A(  _SCREAMING_SNAKE_CASE		:       np.ndarray						,_SCREAMING_SNAKE_CASE		:       int						,_SCREAMING_SNAKE_CASE		:       Optional[bool] = True						,_SCREAMING_SNAKE_CASE		:       Optional[bool] = True						,_SCREAMING_SNAKE_CASE		:       float = 0.0						,)    ->   np.ndarray:


						'''simple docstring'''
						# make sure we normalize float32 arrays
						if normalize_means:
										A	 =      x[:input_length].mean(axis=0		)
										A	 =      np.subtract(_SCREAMING_SNAKE_CASE						,_SCREAMING_SNAKE_CASE		)
						if normalize_vars:
										A	 =      x[:input_length].std(axis=0		)
										A	 =      np.divide(_SCREAMING_SNAKE_CASE						,_SCREAMING_SNAKE_CASE		)

						if input_length < x.shape[0]:
										A	 =      padding_value

						# make sure array is in float32
						A	 =      x.astype(np.floataa		)

						return x
		def       A(  self		:       Any						,_SCREAMING_SNAKE_CASE		:       List[np.ndarray]						,_SCREAMING_SNAKE_CASE		:       Optional[np.ndarray] = None		)    ->   List[np.ndarray]:


						'''simple docstring'''
						A	 =      attention_mask.sum(-1		) if attention_mask is not None else [x.shape[0] for x in input_features]
						return [
						    self.utterance_cmvn(_SCREAMING_SNAKE_CASE						,_SCREAMING_SNAKE_CASE						,self.normalize_means						,self.normalize_vars						,self.padding_value		)
						    for x, n in zip(_SCREAMING_SNAKE_CASE						,_SCREAMING_SNAKE_CASE		)
						]







		def __call__(  self		:       List[str]						,_SCREAMING_SNAKE_CASE		:       Union[np.ndarray, List[float], List[np.ndarray], List[List[float]]]						,_SCREAMING_SNAKE_CASE		:       Union[bool, str, PaddingStrategy] = False						,_SCREAMING_SNAKE_CASE		:       Optional[int] = None						,_SCREAMING_SNAKE_CASE		:       bool = False						,_SCREAMING_SNAKE_CASE		:       Optional[int] = None						,_SCREAMING_SNAKE_CASE		:       Optional[Union[str, TensorType]] = None						,_SCREAMING_SNAKE_CASE		:       Optional[int] = None						,_SCREAMING_SNAKE_CASE		:       Optional[bool] = None						,**_SCREAMING_SNAKE_CASE		:       Dict						,)    ->   BatchFeature:


						'''simple docstring'''

						if sampling_rate is not None:
										if sampling_rate != self.sampling_rate:
														raise ValueError(
														    f'The model corresponding to this feature extractor: {self} was trained using a sampling rate of'
														    f' {self.sampling_rate}. Please make sure that the provided `raw_speech` input was sampled with'
														    f' {self.sampling_rate} and not {sampling_rate}.'		)
						else:
										logger.warning(
										    'It is strongly recommended to pass the `sampling_rate` argument to this function. '
										    'Failing to do so can result in silent errors that might be hard to debug.'		)

						A	 =      isinstance(_SCREAMING_SNAKE_CASE						,np.ndarray		) and len(raw_speech.shape		) > 1
						if is_batched_numpy and len(raw_speech.shape		) > 2:
										raise ValueError(f'Only mono-channel audio is supported for input to {self}'		)
						A	 =      is_batched_numpy or (
						    isinstance(_SCREAMING_SNAKE_CASE						,(list, tuple)		) and (isinstance(raw_speech[0]						,(np.ndarray, tuple, list)		))
						)

						if is_batched:
										A	 =      [np.asarray(_SCREAMING_SNAKE_CASE						,dtype=np.floataa		) for speech in raw_speech]
						elif not is_batched and not isinstance(_SCREAMING_SNAKE_CASE						,np.ndarray		):
										A	 =      np.asarray(_SCREAMING_SNAKE_CASE						,dtype=np.floataa		)
						elif isinstance(_SCREAMING_SNAKE_CASE						,np.ndarray		) and raw_speech.dtype is np.dtype(np.floataa		):
										A	 =      raw_speech.astype(np.floataa		)

						# always return batch
						if not is_batched:
										A	 =      [raw_speech]

						# extract fbank features
						A	 =      [self._extract_fbank_features(_SCREAMING_SNAKE_CASE		) for waveform in raw_speech]

						# convert into correct format for padding
						A	 =      BatchFeature({'input_features': features}		)

						A	 =      self.pad(
						    _SCREAMING_SNAKE_CASE						,padding=_SCREAMING_SNAKE_CASE						,max_length=_SCREAMING_SNAKE_CASE						,truncation=_SCREAMING_SNAKE_CASE						,pad_to_multiple_of=_SCREAMING_SNAKE_CASE						,return_attention_mask=_SCREAMING_SNAKE_CASE						,**_SCREAMING_SNAKE_CASE						,)

						# make sure list is in array format
						A	 =      padded_inputs.get('input_features'		)
						if isinstance(input_features[0]						,_SCREAMING_SNAKE_CASE		):
										A	 =      [np.asarray(_SCREAMING_SNAKE_CASE						,dtype=np.floataa		) for feature in input_features]

						A	 =      padded_inputs.get('attention_mask'		)
						if attention_mask is not None:
										A	 =      [np.asarray(_SCREAMING_SNAKE_CASE						,dtype=np.intaa		) for array in attention_mask]

						# Utterance-level cepstral mean and variance normalization
						if self.do_ceptral_normalize:
										A	 =      (
										    np.array(_SCREAMING_SNAKE_CASE						,dtype=np.intaa		)
										    if self._get_padding_strategies(_SCREAMING_SNAKE_CASE						,max_length=_SCREAMING_SNAKE_CASE		) is not PaddingStrategy.DO_NOT_PAD
										    else None
										)
										A	 =      self.normalize(
										    padded_inputs['input_features']						,attention_mask=_SCREAMING_SNAKE_CASE		)

						if return_tensors is not None:
										A	 =      padded_inputs.convert_to_tensors(_SCREAMING_SNAKE_CASE		)

						return padded_inputs













import unittest

from transformers import JukeboxTokenizer
from transformers.testing_utils import require_torch
class      UpperCamelCase (				unittest.TestCase ):







		"""simple docstring"""



		snake_case    =   JukeboxTokenizer
		snake_case    =   {
		    "artist": "Zac Brown Band",
		    "genres": "Country",
		    "lyrics": "I met a traveller from an antique land,\n        Who said \"Two vast and trunkless legs of stone\n        Stand in the desert. . . . Near them, on the sand,\n        Half sunk a shattered visage lies, whose frown,\n        And wrinkled lip, and sneer of cold command,\n        Tell that its sculptor well those passions read\n        Which yet survive, stamped on these lifeless things,\n        The hand that mocked them, and the heart that fed;\n        And on the pedestal, these words appear:\n        My name is Ozymandias, King of Kings;\n        Look on my Works, ye Mighty, and despair!\n        Nothing beside remains. Round the decay\n        Of that colossal Wreck, boundless and bare\n        The lone and level sands stretch far away\n        ",
		}
		@require_torch
		def       A(  self		:       Optional[int]		)    ->   Any:


						'''simple docstring'''
						import torch

						A	 =      JukeboxTokenizer.from_pretrained('openai/jukebox-1b-lyrics'		)
						A	 =      tokenizer(**self.metas		)['input_ids']
						# fmt: off
						A	 =      [
						    torch.tensor([[
						        0, 0, 0, 7_1_6_9, 5_0_7, 9, 7_6, 3_9, 3_1, 4_6, 7_6, 2_7,
						        7_6, 4_6, 4_4, 2_7, 4_8, 3_1, 3_8, 3_8, 3_1, 4_4, 7_6, 3_2,
						        4_4, 4_1, 3_9, 7_6, 2_7, 4_0, 7_6, 2_7, 4_0, 4_6, 3_5, 4_3,
						        4_7, 3_1, 7_6, 3_8, 2_7, 4_0, 3_0, 6_4, 7_8, 7_6, 7_6, 7_6,
						        7_6, 7_6, 7_6, 7_6, 7_6, 2_3, 3_4, 4_1, 7_6, 4_5, 2_7, 3_5,
						        3_0, 7_6, 7_1, 2_0, 4_9, 4_1, 7_6, 4_8, 2_7, 4_5, 4_6, 7_6,
						        2_7, 4_0, 3_0, 7_6, 4_6, 4_4, 4_7, 4_0, 3_7, 3_8, 3_1, 4_5,
						        4_5, 7_6, 3_8, 3_1, 3_3, 4_5, 7_6, 4_1, 3_2, 7_6, 4_5, 4_6,
						        4_1, 4_0, 3_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
						        1_9, 4_6, 2_7, 4_0, 3_0, 7_6, 3_5, 4_0, 7_6, 4_6, 3_4, 3_1,
						        7_6, 3_0, 3_1, 4_5, 3_1, 4_4, 4_6, 6_3, 7_6, 6_3, 7_6, 6_3,
						        7_6, 6_3, 7_6, 1_4, 3_1, 2_7, 4_4, 7_6, 4_6, 3_4, 3_1, 3_9,
						        6_4, 7_6, 4_1, 4_0, 7_6, 4_6, 3_4, 3_1, 7_6, 4_5, 2_7, 4_0,
						        3_0, 6_4, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 8,
						        2_7, 3_8, 3_2, 7_6, 4_5, 4_7, 4_0, 3_7, 7_6, 2_7, 7_6, 4_5,
						        3_4, 2_7, 4_6, 4_6, 3_1, 4_4, 3_1, 3_0, 7_6, 4_8, 3_5, 4_5,
						        2_7, 3_3, 3_1, 7_6, 3_8, 3_5, 3_1, 4_5, 6_4, 7_6, 4_9, 3_4,
						        4_1, 4_5, 3_1, 7_6, 3_2, 4_4, 4_1, 4_9, 4_0, 6_4, 7_8, 7_6,
						        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1, 4_0, 3_0, 7_6, 4_9,
						        4_4, 3_5, 4_0, 3_7, 3_8, 3_1, 3_0, 7_6, 3_8, 3_5, 4_2, 6_4,
						        7_6, 2_7, 4_0, 3_0, 7_6, 4_5, 4_0, 3_1, 3_1, 4_4, 7_6, 4_1,
						        3_2, 7_6, 2_9, 4_1, 3_8, 3_0, 7_6, 2_9, 4_1, 3_9, 3_9, 2_7,
						        4_0, 3_0, 6_4, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
						        2_0, 3_1, 3_8, 3_8, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_5, 4_6,
						        4_5, 7_6, 4_5, 2_9, 4_7, 3_8, 4_2, 4_6, 4_1, 4_4, 7_6, 4_9,
						        3_1, 3_8, 3_8, 7_6, 4_6, 3_4, 4_1, 4_5, 3_1, 7_6, 4_2, 2_7,
						        4_5, 4_5, 3_5, 4_1, 4_0, 4_5, 7_6, 4_4, 3_1, 2_7, 3_0, 7_8,
						        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 2_3, 3_4, 3_5, 2_9,
						        3_4, 7_6, 5_1, 3_1, 4_6, 7_6, 4_5, 4_7, 4_4, 4_8, 3_5, 4_8,
						        3_1, 6_4, 7_6, 4_5, 4_6, 2_7, 3_9, 4_2, 3_1, 3_0, 7_6, 4_1,
						        4_0, 7_6, 4_6, 3_4, 3_1, 4_5, 3_1, 7_6, 3_8, 3_5, 3_2, 3_1,
						        3_8, 3_1, 4_5, 4_5, 7_6, 4_6, 3_4, 3_5, 4_0, 3_3, 4_5, 6_4,
						        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 2_0, 3_4, 3_1,
						        7_6, 3_4, 2_7, 4_0, 3_0, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_9,
						        4_1, 2_9, 3_7, 3_1, 3_0, 7_6, 4_6, 3_4, 3_1, 3_9, 6_4, 7_6,
						        2_7, 4_0, 3_0, 7_6, 4_6, 3_4, 3_1, 7_6, 3_4, 3_1, 2_7, 4_4,
						        4_6, 7_6, 4_6, 3_4, 2_7, 4_6, 7_6, 3_2, 3_1, 3_0, 6_6, 7_8,
						        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1, 4_0, 3_0, 7_6,
						        4_1, 4_0, 7_6, 4_6, 3_4, 3_1, 7_6, 4_2, 3_1, 3_0, 3_1, 4_5,
						        4_6, 2_7, 3_8, 6_4, 7_6, 4_6, 3_4, 3_1, 4_5, 3_1, 7_6, 4_9,
						        4_1, 4_4, 3_0, 4_5, 7_6, 2_7, 4_2, 4_2, 3_1, 2_7, 4_4, 6_5,
						        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_3, 5_1, 7_6,
						        4_0, 2_7, 3_9, 3_1, 7_6, 3_5, 4_5, 7_6, 1_5, 5_2, 5_1, 3_9,
						        2_7, 4_0, 3_0, 3_5, 2_7, 4_5, 6_4, 7_6, 1_1, 3_5, 4_0, 3_3,
						        7_6, 4_1, 3_2, 7_6, 1_1, 3_5, 4_0, 3_3, 4_5, 6_6, 7_8, 7_6,
						        7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_2, 4_1, 4_1, 3_7, 7_6,
						        4_1, 4_0, 7_6, 3_9, 5_1, 7_6, 2_3, 4_1, 4_4, 3_7, 4_5, 6_4,
						        7_6, 5_1, 3_1, 7_6, 1_3, 3_5, 3_3, 3_4, 4_6, 5_1, 6_4, 7_6,
						        2_7, 4_0, 3_0, 7_6, 3_0, 3_1, 4_5, 4_2, 2_7, 3_5, 4_4, 6_7,
						        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_4, 4_1, 4_6,
						        3_4, 3_5, 4_0, 3_3, 7_6, 2_8, 3_1, 4_5, 3_5, 3_0, 3_1, 7_6,
						        4_4, 3_1, 3_9, 2_7, 3_5, 4_0, 4_5, 6_3, 7_6, 1_8, 4_1, 4_7,
						        4_0, 3_0, 7_6, 4_6, 3_4, 3_1, 7_6, 3_0, 3_1, 2_9, 2_7, 5_1,
						        7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 1_5, 3_2, 7_6,
						        4_6, 3_4, 2_7, 4_6, 7_6, 2_9, 4_1, 3_8, 4_1, 4_5, 4_5, 2_7,
						        3_8, 7_6, 2_3, 4_4, 3_1, 2_9, 3_7, 6_4, 7_6, 2_8, 4_1, 4_7,
						        4_0, 3_0, 3_8, 3_1, 4_5, 4_5, 7_6, 2_7, 4_0, 3_0, 7_6, 2_8,
						        2_7, 4_4, 3_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
						        2_0, 3_4, 3_1, 7_6, 3_8, 4_1, 4_0, 3_1, 7_6, 2_7, 4_0, 3_0,
						        7_6, 3_8, 3_1, 4_8, 3_1, 3_8, 7_6, 4_5, 2_7, 4_0, 3_0, 4_5,
						        7_6, 4_5, 4_6, 4_4, 3_1, 4_6, 2_9, 3_4, 7_6, 3_2, 2_7, 4_4,
						        7_6, 2_7, 4_9, 2_7, 5_1, 7_8, 7_6, 7_6, 7_6, 7_6, 7_6, 7_6,
						        7_6, 7_6]]		),
						    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1]]		),
						    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1]]		),
						]
						# fmt: on
						self.assertTrue(torch.allclose(tokens[0]						,EXPECTED_OUTPUT[0]		)		)
						self.assertTrue(torch.allclose(tokens[1]						,EXPECTED_OUTPUT[1]		)		)
						self.assertTrue(torch.allclose(tokens[2]						,EXPECTED_OUTPUT[2]		)		)







		@require_torch
		def       A(  self		:       Tuple		)    ->   List[Any]:


						'''simple docstring'''
						import torch

						A	 =      JukeboxTokenizer.from_pretrained('openai/jukebox-5b-lyrics'		)
						A	 =      tokenizer(**self.metas		)['input_ids']
						# fmt: off
						A	 =      [
						    torch.tensor([[
						        0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1, 9, 7_7, 3_9,
						        3_1, 4_6, 7_7, 2_7, 7_7, 4_6, 4_4, 2_7, 4_8, 3_1, 3_8, 3_8,
						        3_1, 4_4, 7_7, 3_2, 4_4, 4_1, 3_9, 7_7, 2_7, 4_0, 7_7, 2_7,
						        4_0, 4_6, 3_5, 4_3, 4_7, 3_1, 7_7, 3_8, 2_7, 4_0, 3_0, 6_4,
						        7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 2_3, 3_4, 4_1,
						        7_7, 4_5, 2_7, 3_5, 3_0, 7_7, 7_2, 2_0, 4_9, 4_1, 7_7, 4_8,
						        2_7, 4_5, 4_6, 7_7, 2_7, 4_0, 3_0, 7_7, 4_6, 4_4, 4_7, 4_0,
						        3_7, 3_8, 3_1, 4_5, 4_5, 7_7, 3_8, 3_1, 3_3, 4_5, 7_7, 4_1,
						        3_2, 7_7, 4_5, 4_6, 4_1, 4_0, 3_1, 7_9, 7_7, 7_7, 7_7, 7_7,
						        7_7, 7_7, 7_7, 7_7, 1_9, 4_6, 2_7, 4_0, 3_0, 7_7, 3_5, 4_0,
						        7_7, 4_6, 3_4, 3_1, 7_7, 3_0, 3_1, 4_5, 3_1, 4_4, 4_6, 6_3,
						        7_7, 6_3, 7_7, 6_3, 7_7, 6_3, 7_7, 1_4, 3_1, 2_7, 4_4, 7_7,
						        4_6, 3_4, 3_1, 3_9, 6_4, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1,
						        7_7, 4_5, 2_7, 4_0, 3_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7,
						        7_7, 7_7, 7_7, 8, 2_7, 3_8, 3_2, 7_7, 4_5, 4_7, 4_0, 3_7,
						        7_7, 2_7, 7_7, 4_5, 3_4, 2_7, 4_6, 4_6, 3_1, 4_4, 3_1, 3_0,
						        7_7, 4_8, 3_5, 4_5, 2_7, 3_3, 3_1, 7_7, 3_8, 3_5, 3_1, 4_5,
						        6_4, 7_7, 4_9, 3_4, 4_1, 4_5, 3_1, 7_7, 3_2, 4_4, 4_1, 4_9,
						        4_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 1,
						        4_0, 3_0, 7_7, 4_9, 4_4, 3_5, 4_0, 3_7, 3_8, 3_1, 3_0, 7_7,
						        3_8, 3_5, 4_2, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 4_5, 4_0, 3_1,
						        3_1, 4_4, 7_7, 4_1, 3_2, 7_7, 2_9, 4_1, 3_8, 3_0, 7_7, 2_9,
						        4_1, 3_9, 3_9, 2_7, 4_0, 3_0, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7,
						        7_7, 7_7, 7_7, 7_7, 2_0, 3_1, 3_8, 3_8, 7_7, 4_6, 3_4, 2_7,
						        4_6, 7_7, 3_5, 4_6, 4_5, 7_7, 4_5, 2_9, 4_7, 3_8, 4_2, 4_6,
						        4_1, 4_4, 7_7, 4_9, 3_1, 3_8, 3_8, 7_7, 4_6, 3_4, 4_1, 4_5,
						        3_1, 7_7, 4_2, 2_7, 4_5, 4_5, 3_5, 4_1, 4_0, 4_5, 7_7, 4_4,
						        3_1, 2_7, 3_0, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        2_3, 3_4, 3_5, 2_9, 3_4, 7_7, 5_1, 3_1, 4_6, 7_7, 4_5, 4_7,
						        4_4, 4_8, 3_5, 4_8, 3_1, 6_4, 7_7, 4_5, 4_6, 2_7, 3_9, 4_2,
						        3_1, 3_0, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1, 4_5, 3_1, 7_7,
						        3_8, 3_5, 3_2, 3_1, 3_8, 3_1, 4_5, 4_5, 7_7, 4_6, 3_4, 3_5,
						        4_0, 3_3, 4_5, 6_4, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        7_7, 2_0, 3_4, 3_1, 7_7, 3_4, 2_7, 4_0, 3_0, 7_7, 4_6, 3_4,
						        2_7, 4_6, 7_7, 3_9, 4_1, 2_9, 3_7, 3_1, 3_0, 7_7, 4_6, 3_4,
						        3_1, 3_9, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 4_6, 3_4, 3_1, 7_7,
						        3_4, 3_1, 2_7, 4_4, 4_6, 7_7, 4_6, 3_4, 2_7, 4_6, 7_7, 3_2,
						        3_1, 3_0, 6_6, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        1, 4_0, 3_0, 7_7, 4_1, 4_0, 7_7, 4_6, 3_4, 3_1, 7_7, 4_2,
						        3_1, 3_0, 3_1, 4_5, 4_6, 2_7, 3_8, 6_4, 7_7, 4_6, 3_4, 3_1,
						        4_5, 3_1, 7_7, 4_9, 4_1, 4_4, 3_0, 4_5, 7_7, 2_7, 4_2, 4_2,
						        3_1, 2_7, 4_4, 6_5, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        7_7, 1_3, 5_1, 7_7, 4_0, 2_7, 3_9, 3_1, 7_7, 3_5, 4_5, 7_7,
						        1_5, 5_2, 5_1, 3_9, 2_7, 4_0, 3_0, 3_5, 2_7, 4_5, 6_4, 7_7,
						        1_1, 3_5, 4_0, 3_3, 7_7, 4_1, 3_2, 7_7, 1_1, 3_5, 4_0, 3_3,
						        4_5, 6_6, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 1_2,
						        4_1, 4_1, 3_7, 7_7, 4_1, 4_0, 7_7, 3_9, 5_1, 7_7, 2_3, 4_1,
						        4_4, 3_7, 4_5, 6_4, 7_7, 5_1, 3_1, 7_7, 1_3, 3_5, 3_3, 3_4,
						        4_6, 5_1, 6_4, 7_7, 2_7, 4_0, 3_0, 7_7, 3_0, 3_1, 4_5, 4_2,
						        2_7, 3_5, 4_4, 6_7, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        7_7, 1_4, 4_1, 4_6, 3_4, 3_5, 4_0, 3_3, 7_7, 2_8, 3_1, 4_5,
						        3_5, 3_0, 3_1, 7_7, 4_4, 3_1, 3_9, 2_7, 3_5, 4_0, 4_5, 6_3,
						        7_7, 1_8, 4_1, 4_7, 4_0, 3_0, 7_7, 4_6, 3_4, 3_1, 7_7, 3_0,
						        3_1, 2_9, 2_7, 5_1, 7_9, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7, 7_7,
						        7_7, 1_5, 3_2, 7_7, 4_6, 3_4, 2_7, 4_6, 7_7, 2_9, 4_1, 3_8,
						        4_1, 4_5, 4_5, 2_7, 3_8, 7_7, 2_3, 4_4, 3_1, 2_9, 3_7, 6_4,
						        7_7, 2_8, 4_1, 4_7, 4_0, 3_0, 3_8, 3_1, 4_5, 4_5, 7_7, 2_7,
						        4_0, 3_0, 7_7, 2_8, 2_7, 4_4, 3_1, 7_9, 7_7, 7_7, 7_7, 7_7,
						        7_7, 7_7, 7_7, 7_7, 2_0, 3_4, 3_1, 7_7, 3_8, 4_1, 4_0, 3_1,
						        7_7, 2_7, 4_0, 3_0, 7_7, 3_8, 3_1, 4_8, 3_1, 3_8, 7_7, 4_5,
						        2_7, 4_0, 3_0, 4_5, 7_7, 4_5, 4_6, 4_4, 3_1, 4_6, 2_9, 3_4,
						        7_7, 3_2, 2_7, 4_4, 7_7, 2_7, 4_9, 2_7, 5_1, 7_9, 7_7, 7_7,
						        7_7, 7_7, 7_7, 7_7, 7_7, 7_7]]		),
						    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1]]		),
						    torch.tensor([[0, 0, 0, 1_0_6_9, 1_1, -1, -1, -1, -1]]		),
						]
						# fmt: on
						self.assertTrue(torch.allclose(tokens[0]						,EXPECTED_OUTPUT[0]		)		)
						self.assertTrue(torch.allclose(tokens[1]						,EXPECTED_OUTPUT[1]		)		)
						self.assertTrue(torch.allclose(tokens[2]						,EXPECTED_OUTPUT[2]		)		)





