我正在使用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]
,它返回a
和b
的正确值,但不能返回c
。