Source Code for Module tests.test_buffer

 1  import numpy 
 2   
 3  from glitter import ArrayBuffer, Datatype, uint8, int8, uint16, int16, uint32, int32, float32 
 4   
 5 -def check_buffer(shape, dtype, vrange): 
 6      minval, maxval = vrange 
 7      data = ((maxval - minval) * numpy.random.random(shape) + minval).astype(dtype.as_numpy()) 
 8      buf = ArrayBuffer(data) 
 9      assert (buf.data == data).all(), "data is broken" 
10      assert buf.shape == data.shape, "shape is broken" 
11      assert buf.dtype == Datatype.from_numpy(data.dtype), "dtype is broken" 
12      assert buf._size == data.nbytes, "_size is broken" 
13   
14 -def test_generator(): 
15      shapes = ((4, 4, 4, 4), (4, 4, 4, 3), (4, 16, 8, 3), (5, 4, 4, 3), (5, 5, 5, 3), (6, 6, 6, 3), (7, 13, 5, 3), (1, 1, 3, 3)) 
16      dtypes = (uint8, int8, uint16, int16, uint32, int32, float32) 
17      vranges = ((0, (1<<8)-1), (-1<<7, (1<<7)-1), (0, (1<<16)-1), (-1<<15, (1<<15)-1), (0, (1<<32)-1), (-1<<31, (1<<31)-1), (-10.0, 10.0)) 
18   
19      for shape in shapes: 
20          for dtype, vrange in zip(dtypes, vranges): 
21              yield check_buffer, shape, dtype, vrange 
22