python模拟side_effect或return_value取决于call_count - python

为了测试轮询功能，我想模拟子函数的调用，以便第一次调用它会失败，而第二次调用它会成功。这是它的非常简化的版本:

poll_function(var1):
    value = sub_function(var1)  # First call will return None
    while not value:
        time.sleep(POLLING_INTERVAL)  
        value = sub_function(var1) # A subsequent call will return a string, e.g "data"
    return value

这可能与Mock框架中的mock对象有关吗？我知道Mock对象具有call_count属性，我应该能够以某种方式使用它。

现在，我已经通过创建用于对sub_function()进行猴子补丁的自定义模拟对象解决了该问题，但是我觉得应该有一种更好的，更简洁的方法:

def test_poll():
    class MyMock(object):                                                      

        def __init__(self, *args):                                             
            self.call_count = 0                                                

        def sub_function(self, *args, **kwargs):                             
            if self.call_count > 1:                                            
                return "data"            
            else:                                                              
                self.call_count += 1                                           
                return None  

    my_mock = MyMock()                                                         
    with patch('sub_function', my_mock.sub_function):           
        ok_(poll_function())         

参考方案

如果我正确理解了您的问题，则可以通过setting side_effect to an iterable来解决。对于您的简单情况:

>>> mock_poll = Mock(side_effect=[None, 'data'])
>>> mock_poll()
None
>>> mock_poll()
'data'

如果要允许无限数量的呼叫，请使用itertools cycle和chain函数:

>>> mock_poll = Mock(side_effect=chain(['first'], cycle(['others'])))

多个数组中的numpy计数频率 - python

我有3300个128x128数组（图像），我想计算这些图像中的唯一颜色。np.unique(task, return_counts = True) 可以给我提供独特的颜色和一张图像的计数。但是为了获得所有这3300张图像的表示。我也想获得所有这些图像的独特结果。获得独特的颜色很容易，但是计数却很难。有人可以帮忙吗？ 参考方案 您的问题尚不完全清楚，但我认为您…

我怎样才能从字典的键中算出对象？ - python

我有这本字典：dict={"asset":[("S3","A1"),"S2",("E4","E5"),("E1","S1"),"A6","A8"], "…

Python GPU资源利用 - python

我有一个Python脚本在某些深度学习模型上运行推理。有什么办法可以找出GPU资源的利用率水平？例如，使用着色器，float16乘法器等。我似乎在网上找不到太多有关这些GPU资源的文档。谢谢！ 参考方案 您可以尝试在像Renderdoc这样的GPU分析器中运行pyxthon应用程序。它将分析您的跑步情况。您将能够获得有关已使用资源，已用缓冲区，不同渲染状态上…

Python:图像处理可产生皱纹纸效果 - python

也许很难描述我的问题。我正在寻找Python中的算法，以在带有某些文本的白色图像上创建皱纹纸效果。我的第一个尝试是在带有文字的图像上添加一些真实的皱纹纸图像（具有透明度）。看起来不错，但副作用是文本没有真正起皱。所以我正在寻找更好的解决方案，有什么想法吗？谢谢 参考方案 除了使用透明性之外，假设您有两张相同尺寸的图像，一张在皱纹纸上明亮，一张在白色背景上有深…

Python uuid4，如何限制唯一字符的长度 - python

在Python中，我正在使用uuid4（）方法创建唯一的字符集。但是我找不到将其限制为10或8个字符的方法。有什么办法吗？uuid4()ffc69c1b-9d87-4c19-8dac-c09ca857e3fc谢谢。 参考方案 尝试：x = uuid4() str(x)[:8] 输出："ffc69c1b" Is there a way to…