en/dequeueHead tested

Author: yyl

unit3/pro8.py

@@ -15,6 +15,19 @@ def getTail(self):
   tail = property(
       fget = lambda self: self.getTail())
 
+  def enqueueHead(self, obj):
+    self._list.prepend(obj)
+
+  def dequeueHead(self):
+    assert not self._list.isEmpty
+    return self._list.extract(self._list.head.value)
+
+  def enqueueTail(self, obj):
+    pass
+
+  def dequeueTail(self):
+    pass
+
   def __iter__(self):
     pass
 

unit3/test.py

@@ -78,5 +78,18 @@ def test_init(self):
     self.assertEqual(self.d.head, None)
     self.assertEqual(self.d.tail, None)
 
+  def test_head(self):
+    self.d.enqueueHead(1)
+    self.assertEqual(self.d.head.value, 1)
+    self.assertEqual(self.d.tail.value, 1)
+    self.d.enqueueHead(2)
+    self.d.enqueueHead(3)
+    self.assertEqual(self.d.head.value, 3)
+    self.assertEqual(self.d.tail.value, 1)
+    self.assertEqual(self.d.dequeueHead() , 3)
+    self.assertEqual(self.d.dequeueHead() , 2)
+    self.assertEqual(self.d.dequeueHead() , 1)
+    self.assertRaises(AssertionError, self.d.dequeueHead)
+
 if __name__ == "__main__":
   unittest.main()