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_domain(): ts = TimeSeries() assert ts.domain == Domain([-inf, inf]) ts.domain = None assert ts.domain == Domain([-inf, inf]) ts.domain = [-inf, 5] assert ts.domain == Domain([-inf, 5]) ts.domain = [5, inf] assert ts.domain == Domain([5, inf]) ts.domain = [-inf, inf] assert ts.domain == Domain(-inf, inf) ts.domain = [2, 5] assert ts.domain == Domain([2, 5]) ts.domain = [[2, 5], [9, 10]] assert ts.domain == Domain([[2, 5], [9, 10]]) ts.domain = [[-inf, 1], [2, 5], [9, 10]] assert ts.domain == Domain([[-inf, 1], [2, 5], [9, 10]]) ts.domain = [[2, 5], [9, 10], [11, inf]] assert ts.domain == Domain([[2, 5], [9, 10], [11, inf]]) ts.domain = [[-inf, 1], [2, 5], [9, 10], [11, inf]] assert ts.domain == Domain([[-inf, 1], [2, 5], [9, 10], [11, inf]])
def test_is_data_in_domain(): data1 = [(1, 2), (2, 3), (6, 1), (7, 4)] ts = TimeSeries(data=data1) # this shouldn't work nose.tools.assert_raises(ValueError, setattr, ts, 'domain', [3, 4]) # this should work ts.domain = [0, 8]
def test_set_domain(): ts = TimeSeries() assert ts.domain == Domain([-inf, inf]) ts.domain = None assert ts.domain == Domain([-inf, inf]) ts.domain = [-inf, 5] assert ts.domain == Domain([-inf, 5]) ts.domain = [5, inf] assert ts.domain == Domain([5, inf]) ts.domain = [-inf, inf] assert ts.domain == Domain([-inf, inf]) ts.domain = [2, 5] assert ts.domain == Domain([2, 5]) ts.domain = [[2, 5], [9, 10]] assert ts.domain == Domain([2, 5], [9, 10]) ts = TimeSeries(data=[(1, 2), (2, 3), (6, 1), (8, 4)]) nose.tools.assert_raises(ValueError, setattr, ts, 'domain', [1.5, 7]) nose.tools.assert_raises(ValueError, setattr, ts, 'domain', [0, 7]) nose.tools.assert_raises(ValueError, setattr, ts, 'domain', [1.5, 9])
def test_time_series(): ts = TimeSeries(data=[(1, 2), (2, 3), (6, 1), (8, 4)]) ts.domain = [1, 8.5] nose.tools.assert_raises(KeyError, ts.get, 0) assert ts[1.5] == 2 assert ts[2.4] == 3 assert ts[6] == 1 assert ts[8.5] == 4 nose.tools.assert_raises(KeyError, ts.get, 9) ts[5] = 7 nose.tools.assert_raises(KeyError, ts.set, 9, 10) nose.tools.assert_raises(KeyError, ts.set, 0, 10) nose.tools.assert_raises(ValueError, TimeSeries, data=[(1, 2), (2, 3), (6, 1), (8, 4)], domain=[1.5, 7])