o
    s·¯i°  ã                   @   sD   d dl Z d dlmZ G dd„ dejƒZG dd„ dejƒZdd„ ZdS )	é    N)Údatac                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚDummyDatasetc                 C   s   d| _ d| _d S )Né
   )Úinp_dimÚout_dim©Úself© r	   úM/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/utils/test_utils.pyÚ__init__   s   
zDummyDataset.__init__c                 C   s   dS )Né   r	   r   r	   r	   r
   Ú__len__
   s   zDummyDataset.__len__c                 C   s   t  d| j¡t  d| j¡fS ©Né   )ÚtorchÚrandnr   r   )r   Úidxr	   r	   r
   Ú__getitem__   s   zDummyDataset.__getitem__N©Ú__name__Ú
__module__Ú__qualname__r   r   r   r	   r	   r	   r
   r      s    r   c                   @   s&   e Zd Zddd„Zdd„ Zdd	„ Zd
S )ÚDummyWaveformDataseté   é   é€>  c                 C   s"   || _ || _|| _d| _|| _d S r   )Úinp_len_waveÚout_len_waveÚtotalÚ	inp_n_sigÚ	out_n_sig)r   r   Ún_srcÚlen_waver	   r	   r
   r      s
   
zDummyWaveformDataset.__init__c                 C   s   | j S ©N)r   r   r	   r	   r
   r      s   zDummyWaveformDataset.__len__c                 C   s(   t  | j| j¡}t  | j| j¡}||fS r#   )r   r   r   r   r    r   )r   r   ÚmixedÚsrcsr	   r	   r
   r      s   z DummyWaveformDataset.__getitem__N)r   r   r   r   r	   r	   r	   r
   r      s    
r   c                  C   s,   t j d¡^} }ttt|  d¡ƒƒt|ƒ S )Nú+Ú.)r   Ú__version__ÚsplitÚtupleÚmapÚint)ÚversionÚsuffixr	   r	   r
   Útorch_version_tuple"   s   r/   )r   Útorch.utilsr   ÚDatasetr   r   r/   r	   r	   r	   r
   Ú<module>   s
    