def enableMemoryOptimizer(self, value): """Set enableMemoryOptimizer to the value. @type value: bool @param value: boolean to enable/disable memory optimizer""" return self.stub.EnableMemoryOptimizer(job_pb2.LayerEnableMemoryOptimizerRequest( layer=self.data, value=value), timeout=Cuebot.Timeout)
def enableMemoryOptimizer(self, value): """Enables or disables the memory optimizer. :type value: bool :param value: whether memory optimizer is enabled """ return self.stub.EnableMemoryOptimizer( job_pb2.LayerEnableMemoryOptimizerRequest(layer=self.data, value=value), timeout=Cuebot.Timeout)
def testEnableMemoryOptimizerFalse(self, getStubMock): stubMock = mock.Mock() stubMock.EnableMemoryOptimizer.return_value = job_pb2.LayerEnableMemoryOptimizerResponse() getStubMock.return_value = stubMock layer = opencue.wrappers.layer.Layer( job_pb2.Layer(name=TEST_LAYER_NAME)) layer.enableMemoryOptimizer(False) stubMock.EnableMemoryOptimizer.assert_called_with( job_pb2.LayerEnableMemoryOptimizerRequest(layer=layer.data, value=False), timeout=mock.ANY)