def test_set_interval(): ts = TimeSeries() nose.tools.assert_raises(KeyError, ts.get, 0) nose.tools.assert_raises(KeyError, ts.set_interval, 2, 4, 5) ts[1.2] = 1 ts[3] = 0 ts[6] = 2 assert ts[0] == 1 assert ts[5.5] == 0 assert ts[7] == 2 ts[2:4] = 5 assert ts.items() == [(1.2, 1), (2, 5), (4, 0), (6, 2)] ts[3:5] = 4 assert ts.items() == [(1.2, 1), (2, 5), (3, 4), (5, 0), (6, 2)] tsc = TimeSeries(ts) ts.set_interval(3, 4, 4) assert ts.items() == [(1.2, 1), (2, 5), (3, 4), (4, 4), (5, 0), (6, 2)] tsc.set_interval(3, 4, 4, compact=True) assert tsc.items() == [(1.2, 1), (2, 5), (3, 4), (5, 0), (6, 2)]
def test_set_interval_datetime(): ts = TimeSeries(default=400) ts[datetime(2012, 1, 4, 12)] = 5 ts[datetime(2012, 1, 9, 18)] = 10 ts[datetime(2012, 1, 8):datetime(2012, 1, 10)] = 100 assert ts.items() == [(datetime(2012, 1, 4, 12, 0), 5), (datetime(2012, 1, 8, 0, 0), 100), (datetime(2012, 1, 10, 0, 0), 10)]
def test_set_interval(): ts = TimeSeries() assert ts[0] is None ts.set_interval(2, 4, 5) assert ts[0] is None assert ts[2] == 5 assert ts[3] == 5 assert ts[4] is None assert ts[5] is None ts = TimeSeries() ts[1.2] = 1 ts[3] = 0 ts[6] = 2 assert ts[0] is None assert ts[5.5] == 0 assert ts[7] == 2 ts[2:4] = 5 assert list(ts.items()) == [(1.2, 1), (2, 5), (4, 0), (6, 2)] ts[3:5] = 4 assert list(ts.items()) == [(1.2, 1), (2, 5), (3, 4), (5, 0), (6, 2)] tsc = TimeSeries(ts) ts.set_interval(3, 4, 4) assert list(ts.items()) == [(1.2, 1), (2, 5), (3, 4), (4, 4), (5, 0), (6, 2)] tsc.set_interval(3, 4, 4, compact=True) assert list(tsc.items()) == [(1.2, 1), (2, 5), (3, 4), (5, 0), (6, 2)] tsd = TimeSeries() nose.tools.assert_raises(ValueError, tsd.set_interval, 4, 4, 4)