class TestQueue1(unittest.TestCase):
def setUp(self):
self.emptyQ = OurQueue()
self.q1 = OurQueue([1, 2, 3])
def test_remove1(self):
"""Q4: test calling remove on queue of size >1"""
item = self.q1.remove()
self.assertEqual(item, 1, "did not return 1st item in queue")
self.assertEqual(2, len(self.q1), "queue should have 2 items")
class TestQueue3(unittest.TestCase):
def setUp(self):
self.q1 = OurQueue([1])
def test_remove_size1(self):
"""Q3: test calling remove on queue of size 1"""
res = self.q1.remove()
self.assertEqual(res, 1,
"removing from Queue of size 1 should return 1")
self.assertEqual(len(self.q1), 0,
"removing from Queue of size 1 should leave len==0")
class TestQueue2(unittest.TestCase):
def setUp(self):
self.q1 = OurQueue()
def test_remove_empty(self):
"""Q2: test calling remove on empty queue"""
res = self.q1.remove()
self.assertEqual(res, None,
"removing from empty Queue should return None")
self.assertEqual(len(self.q1), 0,
"removing from empty Queue should leave len==0")
def test_remove_size1():
"""Q3: test calling remove on queue of size 1"""
q1 = OurQueue([1])
res = q1.remove()
assert res == 1
assert len(q1) == 0
def test_remove_empty():
"""Q2: test calling remove on empty queue"""
q1 = OurQueue()
res = q1.remove()
assert res == None, "removing from empty Queue should return None"
assert len(q1) == 0, "removing from empty Queue should leave len==0"