      "\n",
      "class TestHeight(object):\n",
      "\n",
      "    def test_height(self):\n",
      "        root = Node(5)\n",
      "        assert_equal(height(root), 1)\n",
      "        insert(root, 2)\n",
      "        insert(root, 8)\n",
      "        insert(root, 1)\n",
      "        insert(root, 3)\n",
      "        assert_equal(height(root), 3)\n",
      "\n",
      "        print('Success: test_height')\n",
      "\n",
      "\n",
      "def main():\n",
 
</s>
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
      "\n",
      "class TestHeight(object):\n",
      "\n",
      "    def test_height(self):\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
    "        bst = BstHeight(Node(5))\n",
    "        assert_equal(bst.height(bst.root), 1)\n",
    "        bst.insert(2)\n",
    "        bst.insert(8)\n",
    "        bst.insert(1)\n",
    "        bst.insert(3)\n",
    "        assert_equal(bst.height(bst.root), 3)\n",
      "\n",
      "        print('Success: test_height')\n",
      "\n",
      "\n",
      "def main():\n",
keep keep keep keep replace replace replace replace replace replace replace keep keep keep keep keep
 Move tree height to a class
graphs_trees/tree_height/height_solution.ipynb