diff --git a/src/finam/tools/io_helper.py b/src/finam/tools/io_helper.py index 654ca75bcfc7b6b219be786e44105ae11faeb046..a588a5c7a28f8dbbb044c2db1b2028f222239b9d 100644 --- a/src/finam/tools/io_helper.py +++ b/src/finam/tools/io_helper.py @@ -4,13 +4,13 @@ import numpy as np from ..interfaces import IInput, IOutput -def pull_compressed(input, time): +def pull_compressed(io, time): """ Pull compressed data from an Input object. Parameters ---------- - input : IInput + io : IInput The Input object to pull data from. time : :class:`datetime <datetime.datetime>` Simulation time to get the data for. @@ -23,21 +23,21 @@ def pull_compressed(input, time): Raises ------ ValueError - If input is not an IInput instance. + If io is not an IInput instance. """ - if not isinstance(input, IInput): - msg = "pull_compressed: Given input is not an Input object." + if not isinstance(io, IInput): + msg = "pull_compressed: Given io-object is not an IInput instance." raise ValueError(msg) - return input.info.grid.to_compressed(input.pull_data(time)) + return io.info.grid.to_compressed(io.pull_data(time)) -def push_compressed(output, time, data, nodata=np.nan): +def push_compressed(io, time, data, nodata=np.nan): """ Push compressed data to an Output object. Parameters ---------- - output : IOutput + io : IOutput The Output object to push data to. time : :class:`datetime <datetime.datetime>` Simulation time of the data set. @@ -50,9 +50,9 @@ def push_compressed(output, time, data, nodata=np.nan): Raises ------ ValueError - If output is not an IOutput instance. + If io is not an IOutput instance. """ - if not isinstance(output, IOutput): - msg = "push_compressed: Given output is not an Output object." + if not isinstance(io, IOutput): + msg = "push_compressed: Given io-object is not an IOutput instance." raise ValueError(msg) - output.push_data(output.info.grid.from_compressed(data, nodata), time) + io.push_data(io.info.grid.from_compressed(data, nodata), time)