Added set of stacks challenge.

stacks_queues/set_of_stacks/test_set_of_stacks.py

@@ -0,0 +1,35 @@
+from nose.tools import assert_equal
+
+
+class TestSetOfStacks(object):
+    
+    def test_set_of_stacks(self):
+        print('Test: Push on an empty stack')
+        capacity = 2
+        stacks = SetOfStacks(capacity)
+        stacks.push(3)
+
+        print('Test: Push on a non-empty stack')
+        stacks.push(5)
+
+        print('Test: Push on a capacity stack to create a new one')
+        stacks.push('a')
+
+        print('Test: Pop on a one element stack to destroy it')
+        assert_equal(stacks.pop(), 'a')
+
+        print('Test: Pop general case')
+        assert_equal(stacks.pop(), 5)
+        assert_equal(stacks.pop(), 3)
+
+        print('Test: Pop on no elements')
+        assert_equal(stacks.pop(), None)
+        
+        print('Success: test_set_of_stacks')
+
+def main():
+    test = TestSetOfStacks()
+    test.test_set_of_stacks()
+    
+if __name__ == '__main__':
+    main()