def test__rebalance__left2right_delta3(self):
# Prepare unbalanced queue
q = MyQueueViaStacks()
q._left_stack = [4, 3, 2, 1]
q._left_stack_size = 4
q._right_stack = [5]
q._right_stack_size = 1
# Do rebalance
q._rebalance()
self.assertEqual([3, 2, 1], q._left_stack)
self.assertEqual(3, q._left_stack_size)
self.assertEqual([4, 5], q._right_stack)
self.assertEqual(2, q._right_stack_size)
def test__rebalance__right2left(self):
# Prepare unbalanced queue
q = MyQueueViaStacks()
q._left_stack = [1]
q._left_stack_size = 1
q._right_stack = [2, 3, 4]
q._right_stack_size = 3
# Do rebalance
q._rebalance()
self.assertEqual([2, 1], q._left_stack)
self.assertEqual(2, q._left_stack_size)
self.assertEqual([3, 4], q._right_stack)
self.assertEqual(2, q._right_stack_size)