NhanNguyen1309/audio-separation-api
0
1o
2 zj� � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dlm Z d dl3m Z m Z G dd� dej
�Zed kr?e�� dS dS )4� )�Path)�mockN)�transform_stem_zip)�ApiError)�ensure_transform_pitch�ensure_transform_tempoc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )5�AudioTransformTestsc C s� | � td�d� | � td�d� | �t�� }td� W d � n1 s$w Y | � |jjd� | �t�� }td� W d � n1 sEw Y | � |jjd� d S )N�d r � Z
invalid_tempo�
Z
invalid_pitch)� assertEqualr r � assertRaisesr � exception�code)�selfZ tempo_errorZ pitch_error� r �]D:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_audio_transform.py�%test_validates_tempo_and_pitch_ranges
s 6 � 7 �z9AudioTransformTests.test_validates_tempo_and_pitch_rangesc C s: t �� ��}t|�}|d }|d }t�|d��}|�dd� |�dd� W d � n1 s.w Y dtd td8td td d f9d
d�}t j10d|d�� t t|�t|�ddd� W d � n1 s`w Y t�|��}| � |�
� dg� W d � n1 s~w Y W d � d S W d � d S 1 s�w Y d S )N� input.zip�11output.zip�w�12vocals.wav� fake wavz ignored.txts ignore me�13input_path� output_path�_�__�returnc S s t |��t | ��� � d S )N)r Z write_bytesZ14read_bytes)r r r r r r r �fake_transform# s z`AudioTransformTests.test_transform_stem_zip_preserves_expected_wav_names.<locals>.fake_transformz-app.audio_transform.apply_tempo_pitch_to_file)Z side_effect�n � ��
tempo_percent�pitch_semitones)�tempfile�TemporaryDirectoryr �zipfile�ZipFile�writestr�str�float�intr Zpatchr r Znamelist)r �temp_dirZ temp_root� input_zip�15output_zip�archiver r r r �4test_transform_stem_zip_preserves_expected_wav_names s$ 16 � � ��"�zHAudioTransformTests.test_transform_stem_zip_preserves_expected_wav_namesc C s� t �� �Z}t|�d }t|�d }t�|d��}|�dd� W d � n1 s(w Y | �t��}tt |�t |�ddd� W d � n1 sIw Y | �17|j j d � W d � d S 1 saw Y d S )18Nr r r z
../vocals.wavr r r r! Zunsafe_zip_path�
r$ r% r r& r' r( r
r r r) r r r �r r, r- r. r/ �exc_infor r r �,test_transform_stem_zip_rejects_unsafe_paths, s 19 � �"�z@AudioTransformTests.test_transform_stem_zip_rejects_unsafe_pathsc C s� t �� �[}t|�d }t|�d }t�|d��}|�dd� W d � n1 s(w Y | �t��}tt |�t |�dddd � W d � n1 sJw Y | �20|j j d21� W d � d S 1 sbw Y d S ) Nr r r r r r r z2-stem)r" r# Z stem_modeZ stem_missingr1 r2 r r r �8test_transform_stem_zip_requires_expected_stems_for_mode9 s 22 � �"�zLAudioTransformTests.test_transform_stem_zip_requires_expected_stems_for_modeN)�__name__�23__module__� __qualname__r r0 r4 r5 r r r r r s24
r �__main__)Zpathlibr Zunittestr r$ r& Zapp.audio_transformr Z25app.errorsr Zapp.transform_validationr r �TestCaser r6 �mainr r r r �<module> s ; �