NhanNguyen1309/audio-separation-api
0
1o
2 ��j� � @ s� d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dlm Z d dl3m Z m Z m
Z
G dd� dej�Zed krAe�� dS dS )4� )�Path)�mockN)�settings)� JobRegistry)�_collect_expected_stems� _create_zip�_demucs_commandc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SeparationSpeedPolicyTestsc C s� t �� �2}tt|�d�}|jddd�}tj�tdd�� t |�}W d � n1 s*w Y W d � n1 s9w Y | �5d|� | � ||� d�d d � d S )6N��root_dir�2-stem�song.wav�� stem_mode�input_filename� demucs_jobs� z-j� �2�
�tempfile�TemporaryDirectoryr r �7create_jobr �patch�objectr r �assertIn� assertEqual�index��self�temp_dir�registry�job�command� r$ �XD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_separation.py�0test_demucs_command_uses_configured_worker_count
s 89 �� � zKSeparationSpeedPolicyTests.test_demucs_command_uses_configured_worker_countc 10 C s� t �� �J}tt|�d�}|jddd�}tj�tdd��$ tj�tdd�� t |�}W d � n1 s3w Y W d � n1 sBw Y W d � n1 sQw Y | �11d |� | � ||� d �d12 d � | �13d |� | � ||� d �d14 d
� d S )Nr15 z4-stemr
r �demucs_overlapg�������?�
demucs_shiftsr z --overlapr z0.1z--shifts�0r r r$ r$ r% �8test_demucs_command_uses_fast_overlap_and_shift_settings s" 16��17 �� � � 18 zSSeparationSpeedPolicyTests.test_demucs_command_uses_fast_overlap_and_shift_settingsc C s� t �� �W}tt|�d�}|jddd�}t|�d }|�� |d �d� |d �d � t||�}t||� t �19|j ��}d20d � |� � D �}W d � n1 sOw Y W d � n1 s^w Y | �
t|�dd h� | �td
d� |�� D ��� d S )Nr21 r r
r z
demucs-outputz22vocals.wavs vocalsz
no_vocals.wavs instrumentalc S s i | ]}|j |�qS r$ )�filename��.0�infor$ r$ r% �23<dictcomp>5 s zYSeparationSpeedPolicyTests.test_zip_is_stored_without_extra_stem_copy.<locals>.<dictcomp>zinstrumental.wavc s s � | ] }|j tjkV qd S )N)�
compress_type�zipfile�24ZIP_STOREDr, r$ r$ r% � <genexpr>8 s � zXSeparationSpeedPolicyTests.test_zip_is_stored_without_extra_stem_copy.<locals>.<genexpr>)r r r r r �mkdir� write_bytesr r r1 �ZipFile�zip_path�infolistr �set�25assertTrue�all�values)r r r! r" �demucs_stem_dirZstems�archive�infosr$ r$ r% �*test_zip_is_stored_without_extra_stem_copy( s 26 2728 �� � zESeparationSpeedPolicyTests.test_zip_is_stored_without_extra_stem_copyN)�__name__�29__module__� __qualname__r& r* r@ r$ r$ r$ r% r s r �__main__)�pathlibr �unittestr r r1 �30app.configr �app.jobsr �app.separationr r r �TestCaser rA �mainr$ r$ r$ r% �<module> s / �