comparison tests/test_PitchBihist.py @ 39:0e70021f251e branch-tests

some tests on OPmellin

     melody_matrix = pbi.get_melody_matrix(melody)
     n_frames = melody_matrix.shape[1]
     assert np.array_equal(melody_matrix[45, :], np.ones(n_frames))
 
 
+def test_second_frame_decomposition():
+    norigframes = 100
+    nframes, win2, hop2 = pbi.second_frame_decomposition(norigframes)
+    assert nframes == 1
+
+
+def test_second_frame_decomposition():
+    norigframes = np.ceil(pbi.melody_sr * 16.)
+    nframes, win2, hop2 = pbi.second_frame_decomposition(norigframes)
+    # for 16-sec segment with .5 hop size expect ~16 frames round up
+    assert nframes == 17
+
+
 def test_bihist_from_melodia():
     melodia_file = 'data/sample_dataset/Melodia/mel_1_2_1.csv'
     bihist = pbi.bihist_from_melodia(melodia_file, secondframedecomp=False)
     assert bihist.shape == (60, 60)