NhanNguyen1309/audio-separation-api
0
1o
2 �j� � @ sv d dl Z d dlmZ d dlmZmZmZmZmZm Z m3Z4 d dl m Z dd� Z
G dd� de j�Zed kr9e �� dS dS )5� N)�build_fallback_beat_grid)�ChordPostprocessConfig�limit_chord_density_per_bar�postprocess_chords�remove_short_chords�snap_chords_to_beats�validate_chord_timeline�viterbi_smooth_chords)�ChordSegmentRecordc C s dd� | D �S )Nc S s g | ]6}|j |j|jf�qS � )� start_sec�end_sec�chord)�.0r r r �_D:\workspace\nextjs\music_toolkit_projects\audio-separation-api\tests\test_chord_postprocess.py�7<listcomp> s zlabels.<locals>.<listcomp>r )�chordsr r r �labels s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d8� Zd d � Zd
S )�ChordPostprocessTestsc C sV t ddd�t ddd�t ddd�g}g d�}t||dtddd ��}| �t|�d9g� d S ) Nr � �Cg������@�Am� �r � r � r F��enable_viterbi�enable_bar_guardrail)� r r )r10 r r � assertEqualr ��selfr �beats� processedr r r �.test_short_chord_between_same_chord_is_removed s 111213�zDChordPostprocessTests.test_short_chord_between_same_chord_is_removedc C s| t ddd�t ddd�t ddd�t ddd �g}g d14�}t||ddd �}| �t|�d� | �|d jd� | �|d jd� d S )
Nr r r r r r ZEmr �Gr )�max_chords_per_bar�
beats_per_bar�����)r15 r �assertLessEqual�lenr r r
r! r r r �5test_multiple_noisy_chords_inside_one_bar_are_limited! s 16171819�zKChordPostprocessTests.test_multiple_noisy_chords_inside_one_bar_are_limitedc C sN t ddd�t ddd�t ddd�g}g d�}t||d�}| �t|�dd g� d S )20Nr r r r r r& r �r r r )r r r& )r21 r r r r! r r r �+test_duplicate_chords_are_merged_after_snap0 s 222324� zAChordPostprocessTests.test_duplicate_chords_are_merged_after_snapc C s t g d��}| �|g d�� d S )N)r r r r )r r r r )r r )r" �smoothedr r r �(test_viterbi_reduces_single_beat_flicker< s z>ChordPostprocessTests.test_viterbi_reduces_single_beat_flickerc C s� t ddd�t ddd�t ddd�g}g d�}t|d|�}| �|d jd� | �|d jd� t||dd � �D ]\}}| �|j|j� | �|j|j� q6d S )25Nr r r g�������?r r r& )r r r r) )r26 r r r r
�zip�
assertGreater)r" r r# r$ �previous�currentr r r �:test_timeline_validation_repairs_overlaps_and_zero_lengthsA s 272829� �zPChordPostprocessTests.test_timeline_validation_repairs_overlaps_and_zero_lengthsc C sZ t ddd�t ddd�t ddd�g}tddd�}t||dtd d d30��}| �t|�d g� d S ) Nr g�z�G��?r g� ��Q��?r r �x )�bpmFr r- )r31 r r r r r r! r r r �/test_fallback_beat_grid_supports_postprocessingQ s 323334� zEChordPostprocessTests.test_fallback_beat_grid_supports_postprocessingN) �__name__�35__module__� __qualname__r% r, r. r0 r5 r8 r r r r r s r �__main__)�unittest�app.beat_trackingr �app.chord_postprocessr r r r r r r �36app.modelsr37 r �TestCaser r9 �mainr r r r �<module> s $ J �