模拟Numpy结构化数组

Question

我试图弄清楚如何模拟一个numpy结构化数组，但运气不佳。 理想情况下，我想做这样的事情：

from mock import MagicMock
mock_obj = MagicMock()
mock_obj['some']['test']['structure'] = 3
assert 3 ==  mock_obj['some']['test']['structure']

我知道如何使用side_effect模拟单个字典，但是还没有弄清楚如何为任意的嵌套__getitem__或__setitem__函数实现该功能。

编辑：

这里是一些上下文：

def function(self): 
    arr = self.my_structured_array['get']['some']['array']
    #Make decisions based on return value of arr

这样，我可以使用一些垃圾值来模拟对象self.my_structured_array ，以测试其他逻辑。 关键是字典对象实际上是与h5py库绑定的，因此为什么要模拟它。

Answer 1

我相信我找到了解决方案。 似乎有些la脚，但这是我已经可以拿到的壁橱：

from mock import MagicMock
mock_obj = MagicMock()
mock_obj.__getitem__().__getitem__().__getitem__.return_value = 3
assert 3 ==  mock_obj['some']['test']['structure']

我真正看到的唯一问题是它不适用于多个级别。 例如， mock_obj['some']['test']返回一个模拟对象，而不是3。