def test_copy_shm(self): source_manager = cishm.ShmManager(create=True) shape = (5,6) fake_image = numpy.arange(shape[0]*shape[1],dtype=numpy.uint8) fake_image.shape = shape curmsg = source_manager.copy_into_shm(0, fake_image) # copy into shared memory result_new = cishm.get_data_copy(curmsg) assert numpy.allclose( result_new, fake_image)
def test_copy_shm_preallocated(self): source_manager = cishm.ShmManager(create=True) shape = (5,6) fake_image = numpy.arange(shape[0]*shape[1],dtype=numpy.uint8) fake_image.shape = shape curmsg = source_manager.copy_into_shm(0, fake_image) # copy into shared memory result_preallocated = numpy.empty( shape, dtype=numpy.uint8 ) result_preallocated2 = cishm.get_data_copy(curmsg,result_preallocated) assert numpy.allclose( result_preallocated, fake_image ) assert result_preallocated2 is result_preallocated