NhanNguyen1309/audio-separation-api
0
1o
2 �
ja � @ s� d dl Zd dlm mZ d dlZd dlmZ d dlm Z d dl3m Z d dl m
Z
mZmZ G dd� dej�Zedkr@e�� 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_seconds�X �� return_value�song.wav��������@�duration_too_long)r �patchr r r r r r r r r r �"test_duration_limit_is_ten_minutes � � 9 �� �z2ValidationTests.test_duration_limit_is_ten_minutesc C r )Nz+app.chord_validation.probe_duration_secondsr r r r �chord_duration_too_long)r r! r r r r r r r r r r �(test_chord_duration_limit_is_ten_minutes' r# z8ValidationTests.test_chord_duration_limit_is_ten_minutesN)�__name__�10__module__� __qualname__r
r r r" r% r r r r r s 11r �__main__)�builtins� @py_builtins�_pytest.assertion.rewrite� assertion�rewrite�12@pytest_ar�unittestr �13app.errorsr �app.chord_validationr �app.validationr r r �TestCaser r& �mainr r r r �<module> s " ) �