NhanNguyen1309/audio-separation-api
0
1o
2 �Njd � @ sj d dl Z d dl mZ d dlmZ d dlmZ d dlmZmZm Z G dd� de j3�Z e dkr3e �
� dS dS ) � N)�mock)�ApiError)�ensure_chord_duration)�ensure_duration�ensure_supported_extension�ensure_supported_stem_modec @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d4� Zd S ) �ValidationTestsc C s4 | � td�d� | � td�d� | � td�d� d S )Nzsong.mp3z.mp3zmix.WAVz.wavz voice.m4az.m4a)� assertEqualr )�self� r �XD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_validation.py�'test_accepts_supported_audio_extensions5 s z7ValidationTests.test_accepts_supported_audio_extensionsc C �V | � t�� }td� W d � n1 sw Y | �|jjd� | �|jjd� d S )Nz archive.zip�unsupported_file_type� )� assertRaisesr r r � exception�code� status_code�r6 �exc_infor r r �(test_rejects_unsupported_audio_extension �7 8 �z8ValidationTests.test_rejects_unsupported_audio_extensionc C r )Nz9-stemr r )r r r r r r r r r r r �"test_rejects_unsupported_stem_mode r z2ValidationTests.test_rejects_unsupported_stem_modec C � t jddd�� | �td�d� W d � n1 sw Y t jddd��! | �t�� }td� W d � n1 s;w Y W d � n1 sJw Y | �|jjd� d S )Nz%app.validation.probe_duration_secondsi, �� return_value�song.wavg������r@�duration_too_long)r �patchr r r r r r r r r r �#test_duration_limit_is_five_minutes � � 9 �� �z3ValidationTests.test_duration_limit_is_five_minutesc C r )Nz+app.chord_validation.probe_duration_seconds� r r g33333�f@�chord_duration_too_long)r r r r r r r r r r r r �*test_chord_duration_limit_is_three_minutes' r! z:ValidationTests.test_chord_duration_limit_is_three_minutesN)�__name__�10__module__� __qualname__r
r r r r$ r r r r r s 11r �__main__)�unittestr �12app.errorsr �app.chord_validationr �app.validationr r r �TestCaser r% �mainr r r r �<module> s ) �