def test_tree(self): tree = Tree(Category) expected = { (self.medieval.id, "Medieval"): { (self.byzantine_empire.id, "Byzantine Empire"): {}, (self.roman_empire.id, "Roman Empire"): { (self.baden_gulden.id, "Baden gulden"): {}, (self.rhenish_guilder.id, "Rhenish guilder"): {} } }, (self.circulating.id, "Circulating"): { (self.usd.id, "United States dollar"): {}, (self.british_pound.id, "British pound"): {}, (self.libyan_dinar.id, "Libyan dinar"): {} } } self.assertEqual(expected, tree.get_tree())
def categories_to_dropdown_options(): tree = Tree(Category) return list(reversed(flatten(tree.get_tree())))