Я использую 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
.