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])
