def test_ts_slice(): ts = TimeSeries(data=[(1, 2), (2, 3), (6, 1), (8, 4)]) new = ts.slice(1.5, 8.5) assert new.domain == Domain(1.5, 8.5) assert new._d == TimeSeries(data=[(1.5, 2), (2, 3), (6, 1), (8, 4)])._d new = ts.slice(1.5, 8.5, slice_domain=False) assert new.domain == Domain(-inf, inf) assert new._d == TimeSeries(data=[(1.5, 2), (2, 3), (6, 1), (8, 4)])._d ts.domain = [0, 8.5] nose.tools.assert_raises(ValueError, ts.slice, -1, -.5) nose.tools.assert_raises(ValueError, ts.slice, 8.8, 9) new = ts.slice(1.5, 8.5) assert new.domain == Domain(1.5, 8.5) assert new._d == TimeSeries(data=[(1.5, 2), (2, 3), (6, 1), (8, 4)])._d new = ts.slice(-1, 9) assert new.domain == Domain(0, 8.5) assert new._d == TimeSeries(data=[(1, 2), (2, 3), (6, 1), (8, 4)])._d args = [[0, 1.5], [2, 5], [5.5, 7], [8, 9]] ts.domain = args new = ts.slice(1.1, 6) assert new.domain == Domain([[1.1, 1.5], [2, 5], [5.5, 6]]) assert new._d == TimeSeries(data=[(1.1, 2), (2, 3), (6, 1)])._d
def test_slice(): ts = TimeSeries(int) ts[0] = 1 ts[1] = 5 ts[4] = 0 ts[6] = 2 assert ts.slice(0.5, 2.5).items() == [(0.5, 1), (1, 5)] assert ts.slice(1.0, 2.5).items() == [(1.0, 5)] assert ts.slice(-1, 1).items() == [(-1, 1), (0, 1), (1, 5)] assert ts.slice(-1, 0.5).items() == [(-1, 1), (0, 1)] nose.tools.assert_raises(ValueError, ts.slice, 2.5, 0)