如何访问模拟的python对象的ith索引?

时间:2019-07-02 22:41:44

标签: python mocking

我正在使用python的pygrib library,并且我想模拟我编写的一个函数,该函数从一种称为grib文件的天气文件中读取数据。

为此,我需要模拟一个gribmessage对象,该对象具有可模拟某些索引处的值的功能。

因此,如果我将gribmessage分配给变量grib,则需要模拟:

a = grib[0]
b = grib[1]
c = grib.data()

我尝试过 grib_mock.data.return_value = 4,将c设置为4,但是我无法访问索引。我也尝试过grib_mock.__getitem__.side_effect = [0, 1, 2],它返回ab的正确值,但不能返回c

0 个答案:

没有答案