def test_head():
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)
dshead = ds.head()
assert len(dshead.time) == 5
assert ds.time[0] == dshead.time[0]
dshead10 = ds.head(n=10)
assert len(dshead10.time) == 10
def test_head_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)
dshead = ds.head()
assert len(dshead.time) == nt
