def test_tail(): nt = 100 data = [] d = np.zeros([nt, 100, 30]) + 1.0 data.append(d) time = _get_time(nt) items = [ItemInfo("Foo")] ds = Dataset(data, time, items) dstail = ds.tail() assert len(dstail.time) == 5 assert ds.time[-1] == dstail.time[-1] dstail10 = ds.tail(n=10) assert len(dstail10.time) == 10
def test_tail_small_dataset(): nt = 2 data = [] d = np.zeros([nt, 100, 30]) + 1.0 data.append(d) time = _get_time(nt) items = [ItemInfo("Foo")] ds = Dataset(data, time, items) dstail = ds.tail() assert len(dstail.time) == nt