def setUp(self): self.dataShape = (1, 100, 100, 10, 1) self.data = (numpy.random.random(self.dataShape) * 100).astype(int) self.data = numpy.ma.masked_array(self.data, mask=numpy.ma.getmaskarray( self.data), fill_value=self.data.dtype.type(0), shrink=False) self.data[0, :1, :1, 0, 0] = numpy.ma.masked graph = Graph() opProvider = OpArrayPiperWithAccessCount(graph=graph) opProvider.Input.meta.axistags = vigra.defaultAxistags('txyzc') opProvider.Input.meta.has_mask = True opProvider.Input.setValue(self.data) self.opProvider = opProvider opCache = OpSlicedBlockedArrayCache(graph=graph) opCache.Input.connect(opProvider.Output) opCache.innerBlockShape.setValue( ((10, 1, 10, 10, 10), (10, 10, 1, 10, 10), (10, 10, 10, 1, 10))) opCache.outerBlockShape.setValue( ((20, 2, 20, 20, 20), (20, 20, 2, 20, 20), (20, 20, 20, 2, 20))) opCache.fixAtCurrent.setValue(False) self.opCache = opCache
def testCleanup(self): try: CacheMemoryManager().disable() op = OpSlicedBlockedArrayCache(graph=self.opProvider.graph) op.Input.connect(self.opProvider.Output) op.BlockShape.setValue(self.opCache.BlockShape.value) op.fixAtCurrent.setValue(False) x = op.Output[...].wait() op.Input.disconnect() op.cleanUp() r = weakref.ref(op) del op gc.collect() assert r() is None, "OpBlockedArrayCache was not cleaned up correctly" finally: CacheMemoryManager().enable()
def testCleanup(self): try: CacheMemoryManager().disable() op = OpSlicedBlockedArrayCache(graph=self.opProvider.graph) op.Input.connect(self.opProvider.Output) op.BlockShape.setValue(self.opCache.BlockShape.value) op.fixAtCurrent.setValue(False) x = op.Output[...].wait() op.Input.disconnect() op.cleanUp() r = weakref.ref(op) del op gc.collect() assert r() is None, "OpBlockedArrayCache was not cleaned up correctly" finally: CacheMemoryManager().enable()
def setUp(self): self.dataShape = (1,100,100,10,1) self.data = (numpy.random.random(self.dataShape) * 100).astype(numpy.uint32) self.data = self.data.view(vigra.VigraArray) self.data.axistags = vigra.defaultAxistags('txyzc') graph = Graph() opProvider = OpArrayPiperWithAccessCount(graph=graph) opProvider.Input.setValue(self.data) self.opProvider = opProvider opCache = OpSlicedBlockedArrayCache(graph=graph) opCache.Input.connect(opProvider.Output) opCache.BlockShape.setValue( ( (20,2,20,20,20), (20,20,2,20,20), (20,20,20,2,20) ) ) opCache.fixAtCurrent.setValue(False) self.opCache = opCache