def test_flipud(): nt = 2 d = np.random.random([nt, 100, 30]) time = _get_time(nt) items = [ItemInfo("Foo")] ds = Dataset([d], time, items) dsud = ds.copy() dsud.flipud() assert dsud.shape == ds.shape assert dsud["Foo"][0, 0, 0] == ds["Foo"][0, -1, 0]
def test_copy(): nt = 100 d1 = np.zeros([nt, 100, 30]) + 1.5 d2 = np.zeros([nt, 100, 30]) + 2.0 data = [d1, d2] time = _get_time(nt) items = [ItemInfo("Foo"), ItemInfo("Bar")] ds = Dataset(data, time, items) assert len(ds.items) == 2 assert len(ds.data) == 2 assert ds.items[0].name == "Foo" ds2 = ds.copy() ds2.items[0].name = "New name" assert ds2.items[0].name == "New name" assert ds.items[0].name == "Foo"