NhanNguyen1309/audio-separation-api
0
1o
2 ��j� � @ s� d dl Zd dlm mZ d dlmZ d dlm Z d dl3Z4d dlZd dl Z d dl m
Z
d dlmZ d dlmZmZmZ G dd� dej�Zed krNe�� dS dS )5� N)�Path)�mock)�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 | �6d|� | � ||� d�d d � d S )7N��root_dir�2-stem�song.wav�� stem_mode�input_filename� demucs_jobs� z-j� �2�
�tempfile�TemporaryDirectoryr r �8create_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 910 �� � zKSeparationSpeedPolicyTests.test_demucs_command_uses_configured_worker_countc 11 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 | �12d |� | � ||� d �d13 d � | �14d |� | � ||� d �d15 d
� d S )Nr16 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" 17��18 �� � � 19 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 �20|j ��}d21d � |� � D �}W d � n1 sOw Y W d � n1 s^w Y | �
t|�dd h� | �td
d� |�� D ��� d S )Nr22 r r
r z
demucs-outputz23vocals.wavs vocalsz
no_vocals.wavs instrumentalc S s i | ]}|j |�qS r$ )�filename��.0�infor$ r$ r% �24<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�25ZIP_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�26assertTrue�all�values)r r r! r" �demucs_stem_dirZstems�archive�infosr$ r$ r% �*test_zip_is_stored_without_extra_stem_copy( s 27 2829 �� � zESeparationSpeedPolicyTests.test_zip_is_stored_without_extra_stem_copyN)�__name__�30__module__� __qualname__r& r* r@ r$ r$ r$ r% r s r �__main__)�builtins� @py_builtins�_pytest.assertion.rewrite� assertion�rewrite�31@pytest_ar�pathlibr �unittestr r r1 �32app.configr �app.jobsr �app.separationr r r �TestCaser rA �mainr$ r$ r$ r% �<module> s & / �