def test_pop(self):
q = MyQueueViaStacks()
q._right_stack = [1]
q._right_stack_size = 1
val = q.pop()
empty_val = q.pop()
self.assertEqual(1, val)
self.assertEqual(0, q._right_stack_size)
self.assertIsNone(empty_val)
def test_pop__the_only_item_on_the_left(self):
q = MyQueueViaStacks()
q._left_stack = [1]
q._left_stack_size = 1
val = q.pop()
empty_val = q.pop()
self.assertEqual(1, val)
self.assertEqual(0, q._left_stack_size)
self.assertIsNone(empty_val)
def test_pushpop(self):
# Queue 1,2,3,4
q = MyQueueViaStacks()
q.put(2)
q.push(3)
q.put(1)
q.push(4)
self.assertEqual(1, q.get())
self.assertEqual(4, q.pop())
self.assertEqual(2, q.get())
self.assertEqual(3, q.pop())
self.assertIsNone(q.get())
self.assertIsNone(q.pop())