NhanNguyen1309/audio-separation-api
0
1o
2 kMj� � @ sJ d dl Z d dlmZ d dlmZ G dd� de j�Zedkr#e �� dS dS )� N)�3TestClient)�appc @ s e Zd Zdd� Zdd� ZdS )�ApiSurfaceTestsc C s8 t t��d�}| �|jd� | �t|�� �h d�� d S )Nz/health�� > Zsupported_chord_endpointZserviceZ4chord_hostZchord_detection_available�statusZ chord_engineZ demucs_modelZseparation_available)r r �get� assertEqual� status_code�assertGreaterEqual�setZjson��selfZresponse� r �YD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_api_surface.py�.test_health_only_reports_separation_capability s 5�z>ApiSurfaceTests.test_health_only_reports_separation_capabilityc C s t t��d�}| �|jd� d S )Nz/api/audio/transformi� )r r Zpostr r r r r r �.test_audio_transform_endpoint_is_not_available s z>ApiSurfaceTests.test_audio_transform_endpoint_is_not_availableN)�__name__�6__module__� __qualname__r r r r r r r s r �__main__) ZunittestZfastapi.testclientr Zapp.mainr �TestCaser r �mainr r r r �<module> s �