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)
