def test_calling(): a = TimeSeries([1, 1.5, 2, 2.5, 10], [0, 2, -1, 0.5, 0]) assert a.times() == [1, 1.5, 2, 2.5, 10] assert a.values() == [0, 2, -1, 0.5, 0]
def test_interpolation(): a = TimeSeries([0,5,10], [1,2,3]) b = TimeSeries([2.5,7.5], [100, -100]) assert (a.interpolate([1]) == TimeSeries([1],[1.2])) assert (a.interpolate(b.times()) == TimeSeries([2.5,7.5], [1.5, 2.5])) assert (a.interpolate([-100,100]) == TimeSeries([-100,100], [1,3]))
def test_time(): ts = TimeSeries(range(0, 7)) assert np.array_equal(ts.times(), np.array([1, 2, 3, 4, 5, 6, 7]))
def test_interpolation(): a = TimeSeries([0, 5, 10], [1, 2, 3]) b = TimeSeries([2.5, 7.5], [100, -100]) assert (a.interpolate([1]) == TimeSeries([1], [1.2])) assert (a.interpolate(b.times()) == TimeSeries([2.5, 7.5], [1.5, 2.5])) assert (a.interpolate([-100, 100]) == TimeSeries([-100, 100], [1, 3]))
def test_times(): data = [0, 1, 2, 3, 4] #range(0,5) time = [5, 6, 7, 8, 9] #range(5,10) ts = TimeSeries(data, time) assert all(ts.times() == np.array(time))