Mercurial > hg > vampy-host
comparison test/test_frames.py @ 56:3db99d3f2e30
Start Python module, add framer
author | Chris Cannam |
---|---|
date | Wed, 14 Jan 2015 09:52:33 +0000 |
parents | |
children | c444be7637c0 |
comparison
equal
deleted
inserted
replaced
55:5e7bda1d5e9a | 56:3db99d3f2e30 |
---|---|
1 | |
2 import vamp | |
3 import numpy as np | |
4 | |
5 def to_lists(arrs): | |
6 return [list([list(r) for r in f]) for f in arrs] | |
7 | |
8 def test_frames_from_1d_buffer(): | |
9 buf = np.arange(6) | |
10 ff = to_lists(vamp.framesFromArray(buf, 2, 2)) | |
11 assert(ff == [[[0,1]],[[2,3]],[[4,5]]]) | |
12 ff = to_lists(vamp.framesFromArray(buf, 1, 2)) | |
13 assert(ff == [[[0,1]],[[1,2]],[[2,3]],[[3,4]],[[4,5]],[[5,0]]]) | |
14 | |
15 def test_frames_from_2d_buffer(): | |
16 buf = np.array([np.arange(6),np.arange(6,12)]) | |
17 ff = to_lists(vamp.framesFromArray(buf, 2, 2)) | |
18 assert(ff == [[[0,1],[6,7]],[[2,3],[8,9]],[[4,5],[10,11]]]) | |
19 ff = to_lists(vamp.framesFromArray(buf, 1, 2)) | |
20 assert(ff == [[[0,1],[6,7]],[[1,2],[7,8]],[[2,3],[8,9]],[[3,4],[9,10]],[[4,5],[10,11]],[[5,0],[11,0]]]) | |
21 | |
22 |