diff --git a/graphs_trees/bst/bst.py b/graphs_trees/bst/bst.py index 87c1a00..b95fe2f 100644 --- a/graphs_trees/bst/bst.py +++ b/graphs_trees/bst/bst.py @@ -17,14 +17,13 @@ class Bst(object): def insert(self, data): if data is None: - raise Exception('Data cannot be None') + raise TypeError('data cannot be None') if self.root is None: self.root = Node(data) return self.root return self._insert(self.root, data) def _insert(self, node, data): - # Constraint: Assume we are working with valid ints if node is None: return Node(data) if data <= node.data: diff --git a/graphs_trees/bst/bst_solution.ipynb b/graphs_trees/bst/bst_solution.ipynb index fd42d3f..e58af89 100644 --- a/graphs_trees/bst/bst_solution.ipynb +++ b/graphs_trees/bst/bst_solution.ipynb @@ -132,14 +132,13 @@ "\n", " def insert(self, data):\n", " if data is None:\n", - " raise Exception('Data cannot be None')\n", + " raise TypeError('data cannot be None')\n", " if self.root is None:\n", " self.root = Node(data)\n", " return self.root\n", " return self._insert(self.root, data)\n", "\n", " def _insert(self, node, data):\n", - " # Constraint: Assume we are working with valid ints\n", " if node is None:\n", " return Node(data)\n", " if data <= node.data:\n",