def test_iteritems(): data = [0, 1, 2, 3, 4] time = [5, 6, 7, 8, 9] ts = ArrayTimeSeries(time, data) l = ts.iteritems() next(l) assert next(l) == (6, 1)
def test_iter_pair(): ts = ArrayTimeSeries(range(1, 8), range(0, 7)) assert np.array_equal( np.array(list(ts.iteritems())), np.array([(1, 0), (2, 1), (3, 2), (4, 3), (5, 4), (6, 5), (7, 6)]))