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)
