def test_split_merge_simple(self):
ds = dstack([1, 2, 3])
ds.split(1) # split 1 off the top
assert map(list, ds) == [[1, 2], [3]]
ds.merge() # merge it back onto the top
ds.split(2) # split 2 off the top
assert map(list, ds) == [[1], [2, 3]]
def test_append_and_extend(self):
ds = dstack()
ds.append(2)
assert map(list, list(ds)) == [[2]]
ds.append('q', 0)
ds.append('r', 1)
assert map(list, list(ds)) == [['q', 2, 'r']]
ds.extend([3, 4], 0)
ds.extend([5, 6], 1)
assert map(list, list(ds)) == [[4, 3, 'q', 2, 'r', 5, 6]]
def test_pop(self):
ds = dstack([1, 2, 3, 4, 5])
x = ds.pop()
assert x == 5
assert map(list, ds) == [[1, 2, 3, 4]]
ds.split(2)
ds.merge(0)
x = ds.pop(0)
assert x == 3
assert map(list, ds) == [[4, 1, 2]]
def test_split_reverse(self):
ds = dstack([1, 2, 3, 4, 5])
ds.split(4, 1, True) # test Reverse split
assert map(list, ds) == [[1], [5, 4, 3, 2]]
ds.split(3, 1, True) # test Reverse split again
assert map(list, ds) == [[1], [5], [2, 3, 4]]
ds.split(2, 0, True) # test Reverse split off bottom
assert map(list, ds) == [[1], [5], [4], [3, 2]]
ds.split(3, 0, True) # test Reverse too-long split off bottom
assert map(list, ds) == [[1], [5], [4], [], [2, 3]]
ds.split(2, 1, True) # test Reverse exact-length split (off top)
assert map(list, ds) == [[1], [5], [4], [], [], [3, 2]]
ds = dstack([1, 2, 3, 4, 5])
ds.split(6, 1, True) # test Reverse too-long split off top
assert map(list, ds) == [[], [5, 4, 3, 2, 1]]
ds.merge()
ds.split(6, 1, True) # test Reverse too-long split off top
assert map(list, ds) == [[], [1, 2, 3, 4, 5]]
ds.merge()
def test_split_merge_degenerate(self):
ds = dstack([1, 2, 3])
ds.split()
assert map(list, ds) == [[1, 2, 3], []]
ds.merge()
assert map(list, ds) == [[1, 2, 3]]
ds.merge()
assert map(list, ds) == [[1, 2, 3]]
ds.split(4)
assert map(list, ds) == [[], [1, 2, 3]]
ds.split(2)
assert map(list, ds) == [[], [1], [2, 3]]
def test_split_merge_end_0(self):
ds = dstack([1, 2, 3])
ds.split(1, 0) # split 1 off the bottom
assert map(list, ds) == [[2, 3], [1]]
ds.append(4) # put 4 on the top of top
ds.append(5, 0) # put 5 on the bottom of top
assert map(list, ds) == [[2, 3], [5, 1, 4]]
ds.merge(0) # slap it all back onto the bottom
assert map(list, ds) == [[5, 1, 4, 2, 3]]
ds.split(2, 0) # take two off the bottom
assert map(list, ds) == [[4, 2, 3], [5, 1]]
ds.merge(0) # slap it all back onto the bottom
assert map(list, ds) == [[5, 1, 4, 2, 3]]
def test_init(self):
ds = dstack()
assert map(list, list(ds)) == [[]]
ds = dstack([1, 2, 4])
assert map(list, list(ds)) == [[1, 2, 4]]
