     },
     "outputs": [],
     "source": [
      "class StackWithCapacity(Stack):\n",
      "    def __init__(self, top=None, capacity=10):\n",
      "        self.capacity = capacity\n",
      "        self.num_items = 0\n",
      "        super(StackWithCapacity, self).__init__(top)\n",
 
</s> remove     "class Test(object):\n",
 </s> add     "\n",
    "class TestSetOfStacks(object):\n",
    "    \n", </s> remove     "    test = Test()\n",
    "    test.test_set_of_stacks()"
 </s> add     "    main()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Test: Push on an empty stack\n",
      "Test: Push on a non-empty stack\n",
      "Test: Push on a capacity stack to create a new one\n",
      "Test: Pop on a one element stack to destroy it\n",
      "Test: Pop general case\n",
      "Test: Pop on no elements\n",
      "Success: test_set_of_stacks\n"
     ]
    }
   ],
   "source": [
    "%run -i test_set_of_stacks.py"
    "    \n",
     },
     "outputs": [],
     "source": [
      "class StackWithCapacity(Stack):\n",
    "    \n",
      "    def __init__(self, top=None, capacity=10):\n",
      "        self.capacity = capacity\n",
      "        self.num_items = 0\n",
      "        super(StackWithCapacity, self).__init__(top)\n",
keep keep keep add keep keep keep keep
 Added set of stacks challenge. 
stacks_queues/set_of_stacks/set_of_stacks_solution.ipynb