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
