def _leaf_item(toolbox=None): if toolbox is None: toolbox = create_toolboxui_with_project() project_item = View(toolbox, toolbox.project(), toolbox, "View", "A View item", 0.0, 0.0) item = LeafProjectTreeItem(project_item, toolbox) return toolbox, item
def test_category_of_item(self): root = RootProjectTreeItem() category = CategoryProjectTreeItem("category", "category description") root.add_child(category) model = ProjectItemModel(self.toolbox, root) self.assertEqual(model.category_of_item("nonexistent item"), None) project_item = ProjectItem("item", "item description", 0.0, 0.0, self.toolbox.project(), self.toolbox) item = LeafProjectTreeItem(project_item, self.toolbox) category.add_child(item) found_category = model.category_of_item("item") self.assertEqual(found_category.name, category.name)
def test_set_leaf_item_name(self): root = RootProjectTreeItem() model = ProjectItemModel(root) category = CategoryProjectTreeItem("category", "category description") model.insert_item(category) category_index = model.find_category("category") project_item = ProjectItem("item", "item description", 0.0, 0.0, self.toolbox.project()) leaf = LeafProjectTreeItem(project_item, self.toolbox) model.insert_item(leaf, category_index) leaf_index = model.find_item("item") model.set_leaf_item_name(leaf_index, "new view item name") leaf_item = model.get_item("new view item name") self.assertIsNotNone(leaf_item)
def test_insert_item_leaf_item(self): root = RootProjectTreeItem() model = ProjectItemModel(self.toolbox, root) category = CategoryProjectTreeItem("category", "category description") model.insert_item(category) category_index = model.find_category("category") mock_project_item = NonCallableMagicMock() mock_project_item.name = "project item" mock_project_item.description = "project item description" leaf = LeafProjectTreeItem(mock_project_item, self.toolbox) model.insert_item(leaf, category_index) self.assertEqual(model.rowCount(), 1) self.assertEqual(model.rowCount(category_index), 1) self.assertEqual(model.n_items(), 1) self.assertEqual(model.items("category"), [leaf])
def _leaf_item(toolbox): project_item = ProjectItem("PI", "A Project item", 0.0, 0.0, toolbox.project()) item = LeafProjectTreeItem(project_item, toolbox) return item
def _leaf_item(toolbox=None): if toolbox is None: toolbox = create_toolboxui_with_project() project_item = ProjectItem("PI", "A Project item", 0.0, 0.0, toolbox.project(), toolbox) item = LeafProjectTreeItem(project_item, toolbox) return toolbox, item