  
  
  class TestCheckBalance(object):
  
      def test_check_balance(self):
          assert_equal(check_balance(None), False)
  
          node = Node(5)
          assert_equal(check_balance(node), True)
  
          insert(node, 3)
 
</s> remove     "    def test_check_balance(self):\n",
    "        assert_equal(check_balance(None), False)\n",
 </s> add     "    @raises(Exception)\n",
    "    def test_check_balance_empty(self):\n",
    "        check_balance(None)\n", </s> remove     "    def test_check_balance(self):\n",
    "        assert_equal(check_balance(None), False)\n",
 </s> add     "    @raises(Exception)\n",
    "    def test_check_balance_empty(self):\n",
    "        check_balance(None)\n",
    @raises(Exception)
    def test_check_balance_empty(self):
        check_balance(None)
  
  
  class TestCheckBalance(object):
  
    @raises(Exception)
    def test_check_balance_empty(self):
        check_balance(None)
    @raises(Exception)
    def test_check_balance_empty(self):
        check_balance(None)
  
          node = Node(5)
          assert_equal(check_balance(node), True)
  
          insert(node, 3)
keep keep keep keep replace replace keep keep keep keep keep
 Polish bst check balanced challenge and solution (#84)

Update constraints, tests, and code. 
graphs_trees/check_balance/test_check_balance.py