Ejemplo n.º 1
0
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
Ejemplo n.º 2
0
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)