class TreeDataTests(unittest.TestCase): def setUp(self): self.win = gtk.Window() self.win.set_default_size(400, 400) self.tree = ObjectTree([Column('name'), Column('age')]) self.win.add(self.tree) refresh_gui() def tearDown(self): self.win.destroy() del self.win def testGetRoot(self): root = Person('Big Kahuna', 7000) child1 = Person('Craf Kahuna', 200) child2 = Person('Sorcerer Kahuna', 150) self.tree.append(None, root) self.tree.append(root, child1) self.tree.append(root, child2) test_root = self.tree.get_root(child1) self.assertEqual(test_root, root) test_root = self.tree.get_root(child2) self.assertEqual(test_root, root) test_root = self.tree.get_root(root) self.assertEqual(test_root, root) def testGetDescendants(self): root = Person('Big Kahuna', 7000) child1 = Person('Craf Kahuna', 200) child2 = Person('Sorcerer Kahuna', 150) self.tree.append(None, root) self.tree.append(root, child1) self.tree.append(child1, child2) test_descendants = self.tree.get_descendants(root) test_descendants.sort() self.assertEqual(test_descendants, [child1, child2]) test_descendants = self.tree.get_descendants(child1) self.assertEqual(test_descendants, [child2]) test_descendants = self.tree.get_descendants(child2) self.assertEqual(test_descendants, [])
class TreeDataTests(unittest.TestCase): def setUp(self): self.win = Gtk.Window() self.win.set_default_size(400, 400) self.tree = ObjectTree([Column('name'), Column('age')]) self.win.add(self.tree) refresh_gui() def tearDown(self): self.win.destroy() del self.win def testGetRoot(self): root = Person('Big Kahuna', 7000) child1 = Person('Craf Kahuna', 200) child2 = Person('Sorcerer Kahuna', 150) self.tree.append(None, root) self.tree.append(root, child1) self.tree.append(root, child2) test_root = self.tree.get_root(child1) self.assertEqual(test_root, root) test_root = self.tree.get_root(child2) self.assertEqual(test_root, root) test_root = self.tree.get_root(root) self.assertEqual(test_root, root) def testGetDescendants(self): root = Person('Big Kahuna', 7000) child1 = Person('Craf Kahuna', 200) child2 = Person('Sorcerer Kahuna', 150) self.tree.append(None, root) self.tree.append(root, child1) self.tree.append(child1, child2) test_descendants = self.tree.get_descendants(root) self.assertTrue(child1 in test_descendants) self.assertTrue(child2 in test_descendants) test_descendants = self.tree.get_descendants(child1) self.assertEqual(test_descendants, [child2]) test_descendants = self.tree.get_descendants(child2) self.assertEqual(test_descendants, [])