Mercurial > hg > vampy-host

comparison test/test_frames.py @ 82:a11b57e9fb0b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
naming: module methods snake_case
author Chris Cannam
date Wed, 21 Jan 2015 12:43:50 +0000
parents c444be7637c0
children 3e5791890b65
comparison
equal deleted inserted replaced
81:0a2f2e7803ea 82:a11b57e9fb0b
5 def to_lists(arrs): 5 def to_lists(arrs):
6 return [list([list(r) for r in f]) for f in arrs] 6 return [list([list(r) for r in f]) for f in arrs]
7 7
8 def test_frames_from_1d_buffer(): 8 def test_frames_from_1d_buffer():
9 buf = np.arange(6) 9 buf = np.arange(6)
10 ff = to_lists(vamp.framesFromArray(buf, 2, 2)) 10 ff = to_lists(vamp.frames_from_array(buf, 2, 2))
11 assert(ff == [[[0,1]],[[2,3]],[[4,5]]]) 11 assert(ff == [[[0,1]],[[2,3]],[[4,5]]])
12 ff = to_lists(vamp.framesFromArray(buf, 1, 2)) 12 ff = to_lists(vamp.frames_from_array(buf, 1, 2))
13 assert(ff == [[[0,1]],[[1,2]],[[2,3]],[[3,4]],[[4,5]],[[5,0]]]) 13 assert(ff == [[[0,1]],[[1,2]],[[2,3]],[[3,4]],[[4,5]],[[5,0]]])
14 14
15 def test_frames_from_2d_buffer(): 15 def test_frames_from_2d_buffer():
16 buf = np.array([np.arange(6),np.arange(6,12)]) 16 buf = np.array([np.arange(6),np.arange(6,12)])
17 ff = to_lists(vamp.framesFromArray(buf, 2, 2)) 17 ff = to_lists(vamp.frames_from_array(buf, 2, 2))
18 assert(ff == [[[0,1],[6,7]],[[2,3],[8,9]],[[4,5],[10,11]]]) 18 assert(ff == [[[0,1],[6,7]],[[2,3],[8,9]],[[4,5],[10,11]]])
19 ff = to_lists(vamp.framesFromArray(buf, 1, 2)) 19 ff = to_lists(vamp.frames_from_array(buf, 1, 2))
20 assert(ff == [[[0,1],[6,7]],[[1,2],[7,8]],[[2,3],[8,9]], 20 assert(ff == [[[0,1],[6,7]],[[1,2],[7,8]],[[2,3],[8,9]],
21 [[3,4],[9,10]],[[4,5],[10,11]],[[5,0],[11,0]]]) 21 [[3,4],[9,10]],[[4,5],[10,11]],[[5,0],[11,0]]])
22 22
23 23