  
  
  class TestCheckBalance(object):
  
      @raises(Exception)
      def test_check_balance_empty(self):
          bst = BstBalance(None)
          bst.check_balance()
  
      def test_check_balance(self):
 
</s> remove     "    @raises(Exception)\n",
 </s> add     "    @raises(TypeError)\n", </s> remove     "    @raises(Exception)\n",
 </s> add     "    @raises(TypeError)\n", </s> remove     "            raise Exception('No root node')\n",
 </s> add     "            raise TypeError('root cannot be None')\n",
    @raises(TypeError)
  
  
  class TestCheckBalance(object):
  
    @raises(TypeError)
      def test_check_balance_empty(self):
          bst = BstBalance(None)
          bst.check_balance()
  
      def test_check_balance(self):
keep keep keep keep replace keep keep keep keep keep
 Update tree check balance challenge (#120)

Use more specific exception types. 
graphs_trees/check_balance/test_check_balance.py