NhanNguyen1309/audio-separation-api
0
1o
2 �Lj � @ s� d dl Z d dlmZ d dlmZmZ d dlmZmZm Z d dl3m Z d dl m
Z
d dlmZmZ G dd � d e j�Zed4krCe �� dS dS ) � N)�build_visual_grid_metadata)� BpmCandidate�choose_best_bpm_candidate)�estimate_bar_start_offset�estimate_downbeats�build_realigned_downbeats)� build_visual_grid_from_beats)�align_chords_to_visual_grid)�ChordSegmentRecord�VisualGridCellRecordc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d5� Zd d � Zd
d� Z dd� Z6dS )�SmartBeatGridTestsc C sL t dddd�}| �|jd� | �t|j�dk� | j|jd jddd� d S )N�I � � @)�bpm�
beats_per_bar� duration_secr � )�places)r � assertEqualZvisual_cells_per_bar�7assertTrue�len�cells�assertAlmostEqual� start_sec)�self�metadata� r �]D:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_smart_beat_grid.py�2test_visual_grid_metadata_keeps_four_cells_per_bar s zESmartBeatGridTests.test_visual_grid_metadata_keeps_four_cells_per_barc C sB t dddd�t dddd�t dd d8d�g}t|�}| �|jd� d S ) N� g9ףp=10�?�tempo�r �score�source�P g�p=11ף�?�halfr
g)\���(�?Zautocorrelation�r r r r �r Z12candidatesZselectedr r r �@test_candidate_scoring_can_choose_seventy_three_over_double_time s �zSSmartBeatGridTests.test_candidate_scoring_can_choose_seventy_three_over_double_timec C s6 t dddd�t dddd�g}t|�}| �|jd� d S )Nr g�������?r! r" r% gH�z�G�?r& r' r( r r r �Dtest_candidate_scoring_prefers_lower_tempo_when_double_time_is_close s13 �zWSmartBeatGridTests.test_candidate_scoring_prefers_lower_tempo_when_double_time_is_closec C sJ g d�}g d�}t |g |dddd�\}}}| �|d� | �|d d� d S )N)r � �?� �?g �?� @g @) r r �14 r r r r r � r r r � )�sr�15hop_lengthr r+ r )r r )r �beats�onset_envelope�offset� downbeats�_r r r �@test_downbeat_offset_can_start_after_zero_when_onset_is_stronger( s16 zSSmartBeatGridTests.test_downbeat_offset_can_start_after_zero_when_onset_is_strongerc C s� g d�}g d�}t ||dd�}| �t|�d� | �|d jd� | �|d jd � | �|d j� | �|d j� | �|d17 j� d S ) N) r r, r- � @� @� @� @� @r )r r: r r �r r/ r r r r: r0 )r r r r r � is_bar_start� assertFalse�r r3 r6 r r r r �!test_build_visual_grid_from_beats1 s z4SmartBeatGridTests.test_build_visual_grid_from_beatsc C s| g d�}g d�}t ||dd�}| �|d jd� | �|d jd� | �|d jd18� | �|d jd � | �|d jd� d S )
N) r r, r- r9 r: r; r< r= �ffffff @)r r: rC rC r> r r: � g333333@� gffffff@� g������ @)r r r �end_secrA r r r �test_variable_bar_duration< s z-SmartBeatGridTests.test_variable_bar_durationc 19 C sR g d�}t ddd�t ddd�g}t|d|dgd d d20d21d d �}| �|d d� d S )
N)22r r: r g (@g 0@g 4@g 8@g <@皙���@@皙���B@r rI �CrJ �Gr, �� r r0 r/ )r3 Zinitial_downbeat_offset�chordsr4 r1 r2 r Zbars_per_segment)r23 r r )r r3 rN r6 r r r � test_realignment_snaps_downbeatF s �24z2SmartBeatGridTests.test_realignment_snaps_downbeatc C s� t ddddddd�t ddddddd�g}td d25d d �td26dd
d �g}t||dd�}| �|d jd� | �|d jd� | �|d jd� | �|d jd� d S )Nr0 r r, T)�indexZvisual_bar_indexZ cell_in_barr rG r? � r- Fg�������?g�������?rK )r rG ZchordrL r> r )r r27 r r r rG )r r rN Zalignedr r r �'test_chord_alignment_maps_to_cell_startU s � �z:SmartBeatGridTests.test_chord_alignment_maps_to_cell_startN) �__name__�28__module__� __qualname__r r) r* r8 rB rH rO rR r r r r r s 29 30 r �__main__)ZunittestZ
app.beat_gridr Zapp.bpm_candidatesr r Z app.downbeatr r r Zapp.visual_gridr Zapp.chord_postprocessr Z31app.modelsr32 r �TestCaser rS �mainr r r r �<module> s Z �