Renamed EmptyStack exception to make more sense and added non functioning unit tests for now.

nryoung · nryoung · commit 780331154c76 · 2013-03-10T23:36:27.000-06:00
diff --git a/algorithms/data_structures/stack.py b/algorithms/data_structures/stack.py
@@ -2,7 +2,7 @@
 Stacks are cool
 """
 
-class EmptyStackException(Exception):
+class EmptyStackError(Exception):
     pass
 
 class Element(object):
@@ -20,7 +20,7 @@ def push(self, element):
 
     def pop(self):
         if self.empty():
-            raise EmptyStackException
+            raise EmptyStackError
         result = self.head.value
         self.head = self.head.next
         return result
diff --git a/algorithms/tests/test_data_structures.py b/algorithms/tests/test_data_structures.py
@@ -0,0 +1,24 @@
+import unittest
+from ..data_structures import Stack, EmptyStackError
+
+class TestStack(unittest.TestCase):
+    """
+    Tests the methods of the Stack data structure.
+    """
+
+    def setUp(self):
+        self.input = range(5)
+        self.stack = Stack()
+        self.reverse = [5, 4, 3, 2, 1]
+
+    def test_stack(self):
+        for i in self.input:
+            self.stack.push(i)
+
+        result = []
+
+        while not self.stack.empty():
+            result.append(self.stack.pop())
+
+        self.assertEqual(self.reverse, result)
+        self.assertRaises(EmptyStackError, self.stack.empty())
