comparison tests/test_load_features.py @ 8:0f3eba42b425 branch-tests

added notebooks and utils

                                         np.arange(np.round(sr * (11.5 - half_win_sec)), 
                                             np.round(sr * (16.5 - half_win_sec)), dtype=int)])
     assert np.array_equal(music_bounds, music_bounds_true)
 
 
+def test_average_local_frames():
+    frames = np.array([[0, 0.5, 1], [1, 1, 1]])
+    aveframes = feat_loader.average_local_frames(frames)
+    aveframes_true = np.array([[0.5], [1]])
+    assert np.array_equal(aveframes, aveframes_true)
+
+
+def test_average_local_frames_multiple_frames():
+    frames1 = np.concatenate([np.repeat(0.5, feat_loader.hop2), np.repeat(0, feat_loader.win2)])
+    frames2 = np.concatenate([np.repeat(1.5, feat_loader.hop2), np.repeat(1, feat_loader.win2)])
+    frames = np.vstack([frames1, frames2])
+    aveframes = feat_loader.average_local_frames(frames)
+    aveframes_true = np.array([[0.5, 0], [1.5, 1]])
+    # test only the second frame which contains values 0 or values 1 for all 8-second frame entries
+    assert np.array_equal(aveframes[:, 1], aveframes_true[:, 1])
+
+
+def test_average_local_frames_std():
+    frames1 = np.concatenate([np.repeat(0.5, feat_loader.hop2), np.repeat(0, feat_loader.win2)])
+    frames2 = np.concatenate([np.repeat(1.5, feat_loader.hop2), np.repeat(1, feat_loader.win2)])
+    frames = np.vstack([frames1, frames2])
+    aveframes = feat_loader.average_local_frames(frames, getstd=True)
+    aveframes_true = np.array([[0.5, 0], [1.5, 1], [0.1, 0], [0.1, 0]])
+    # test only the second frame which contains values 0 or values 1 for all 8-second frame entries
+    assert np.array_equal(aveframes[:, 1], aveframes_true[:, 1])