o
    }oid                     @   s   d dl Z d dlZd dlZd dlZd dlmZ e dd Ze dd Z	e dd Z
e d	d
 Ze dd Ze dd ZdS )    N)read_manifestc                   C   s   t t j r
dS dS )Nzcuda:0cpu)torchdevicecudais_available r   r   R/home/ubuntu/.local/lib/python3.10/site-packages/tests/collections/tts/conftest.py
set_device   s   r
   c                   C   s   dddddS )Nz:Caslon's type is clear and neat, and fairly well designed;u*   Ich trinke gerne Kräutertee mit Lavendel.z'Los corazones de pollo son una delicia.u1   双辽境内除东辽河、西辽河等5条河流)endeeszhr   r   r   r   r	   language_specific_text_example   s
   r   c                 C   s   t |  S )N)sortedkeys)r   r   r   r	   supported_languages'   s   r   c                    s    fdd}|S )Nc                    s2   |  dd }| vrtd|  d  d |S )N_   z+`PretrainedModelInfo.pretrained_model_name=zi` does not follow the naming convention as `tts_languageID_model_*`, or `languageID` is not supported in .)splitpytestfail)pretrained_model_namelanguage_idr   r   r	   	_validate.   s   z=get_language_id_from_pretrained_model_name.<locals>._validater   )r   r   r   r   r	   *get_language_id_from_pretrained_model_name,   s   	r   c                 C   s6   d}d}d}d}d}|| t j|||| d | }|S )Ng      &g      ?r   P   iJ  r   )r   rand)r
   min_valmax_val
batch_sizen_mel_channelsn_framesspecr   r   r	   mel_spec_example:   s   r'   c           
      C   s|   t j| d}t|}|d d }|d d }t|\}}tj|tj|d	d}tj|j
d |d 	d}	||	|fS )Nztts/mini_ljspeech/manifest.jsonaudio_filepathtext)dtyper   r   r   )ospathjoinr   sfreadr   tensorfloat	unsqueezeshapelong)
test_data_dirr
   manifest_pathdatar)   text_raw
audio_dataorig_sraudio	audio_lenr   r   r	   audio_text_pair_example_englishG   s   
r>   )r,   r   	soundfiler/   r   /nemo.collections.asr.parts.utils.manifest_utilsr   fixturer
   r   r   r   r'   r>   r   r   r   r	   <module>   s"   

	


