def test_increment_age_category(self): tree = Tree(None, 0, Tree.sapling) tree.increase_age(12) self.assertEqual(tree.tree_type, Tree.normal)