Add some test to category_tree helper method