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)