def test_add_tree(self, mock_get_float, mock_get_string):
     controller.add_tree()
     oak_tree = Tree.get_or_none(Tree.name == 'oak'
                                 and Tree.max_height == 50)
     self.assertIsNotNone(oak_tree)
 def test_add_tree(self):
     database.add_tree('oak', 50)
     oak_tree = Tree.get_or_none(Tree.name=='oak' and Tree.max_height==50)
     self.assertIsNotNone(oak_tree)