|
Chris@56
|
1
|
|
Chris@95
|
2 import vamp.frames as fr
|
|
Chris@56
|
3 import numpy as np
|
|
Chris@56
|
4
|
|
Chris@56
|
5 def to_lists(arrs):
|
|
Chris@56
|
6 return [list([list(r) for r in f]) for f in arrs]
|
|
Chris@56
|
7
|
|
Chris@56
|
8 def test_frames_from_1d_buffer():
|
|
Chris@56
|
9 buf = np.arange(6)
|
|
Chris@95
|
10 ff = to_lists(fr.frames_from_array(buf, 2, 2))
|
|
Chris@56
|
11 assert(ff == [[[0,1]],[[2,3]],[[4,5]]])
|
|
Chris@95
|
12 ff = to_lists(fr.frames_from_array(buf, 1, 2))
|
|
Chris@56
|
13 assert(ff == [[[0,1]],[[1,2]],[[2,3]],[[3,4]],[[4,5]],[[5,0]]])
|
|
Chris@56
|
14
|
|
Chris@56
|
15 def test_frames_from_2d_buffer():
|
|
Chris@56
|
16 buf = np.array([np.arange(6),np.arange(6,12)])
|
|
Chris@95
|
17 ff = to_lists(fr.frames_from_array(buf, 2, 2))
|
|
Chris@56
|
18 assert(ff == [[[0,1],[6,7]],[[2,3],[8,9]],[[4,5],[10,11]]])
|
|
Chris@95
|
19 ff = to_lists(fr.frames_from_array(buf, 1, 2))
|
|
Chris@58
|
20 assert(ff == [[[0,1],[6,7]],[[1,2],[7,8]],[[2,3],[8,9]],
|
|
Chris@58
|
21 [[3,4],[9,10]],[[4,5],[10,11]],[[5,0],[11,0]]])
|
|
Chris@56
|
22
|
|
Chris@56
|
23
|
