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