def test_itertimes(): data = [0, 1, 2, 3, 4] #range(0,5) time = [5, 6, 7, 8, 9] #range(5,10) ts = ArrayTimeSeries(time, data) l = ts.itertimes() next(l) assert next(l) == 6
def test_valid_input_interpolate(): ts_1 = ArrayTimeSeries([1, 2, 3], [0, 5, 10]) ts_2 = ArrayTimeSeries([100, -100], [2.5, 7.5]) ts_interpolate_test_1 = ts_1.interpolate([1]) assert ts_interpolate_test_1._value == [1.2] assert ts_interpolate_test_1._time == [1] ts_interpolate_test_2 = ts_1.interpolate([-100, 100]) assert ts_interpolate_test_2 == ArrayTimeSeries([1, 3], [-100, 100]) ts_interpolate_test_3 = ts_1.interpolate([2.5, 7.5]) assert ts_interpolate_test_3 == ArrayTimeSeries([1.5, 2.5], [2.5, 7.5]) assert ts_1.interpolate(ts_2.itertimes()) == ArrayTimeSeries([1.5, 2.5], [2.5, 7.5])
def test_iter_time(): ts = ArrayTimeSeries(range(1, 8), range(0, 7)) assert list(ts.itertimes()) == [1, 2, 3, 4, 5, 6, 7]