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