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_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_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))
