This commit was manufactured by cvs2svn to create tag 'WX_2_8_3'.
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/tags/WX_2_8_3@44921 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
Bryan Petty
363
wxPython/tests/TreeMixinTest.py
Normal file
363
wxPython/tests/TreeMixinTest.py
Normal file
@@ -0,0 +1,363 @@
|
||||
import wx, wx.gizmos, wx.lib.customtreectrl, unittest
|
||||
from wx.lib.mixins import treemixin
|
||||
|
||||
|
||||
# VirtualTree tests
|
||||
|
||||
class VirtualTreeCtrl(treemixin.VirtualTree, wx.TreeCtrl):
|
||||
def __init__(self, *args, **kwargs):
|
||||
self.children = {}
|
||||
super(VirtualTreeCtrl, self).__init__(*args, **kwargs)
|
||||
|
||||
def OnGetItemText(self, indices):
|
||||
return 'item %s'%'.'.join([str(index) for index in indices])
|
||||
|
||||
def OnGetChildrenCount(self, indices=None):
|
||||
indices = indices or ()
|
||||
return self.children.get(indices, 0)
|
||||
|
||||
def SetChildrenCount(self, indices, childrenCount):
|
||||
self.children[tuple(indices)] = childrenCount
|
||||
|
||||
|
||||
class VirtualTreeCtrlTest_NoRootItems(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
self.tree = VirtualTreeCtrl(self.frame)
|
||||
self.tree.RefreshItems()
|
||||
|
||||
def testNoRootItems(self):
|
||||
self.assertEqual(0, self.tree.GetCount())
|
||||
|
||||
def testAddTwoRootItems(self):
|
||||
self.tree.SetChildrenCount((), 2)
|
||||
self.tree.RefreshItems()
|
||||
self.assertEqual(2, self.tree.GetCount())
|
||||
|
||||
def testAddOneRootItemAndOneChild(self):
|
||||
self.tree.SetChildrenCount((), 1)
|
||||
self.tree.SetChildrenCount((0,), 1)
|
||||
self.tree.RefreshItems()
|
||||
self.tree.ExpandAll()
|
||||
self.assertEqual(2, self.tree.GetCount())
|
||||
|
||||
def testAddOneRootItemAndTwoChildren(self):
|
||||
self.tree.SetChildrenCount((), 1)
|
||||
self.tree.SetChildrenCount((0,), 2)
|
||||
self.tree.RefreshItems()
|
||||
self.tree.ExpandAll()
|
||||
self.assertEqual(3, self.tree.GetCount())
|
||||
|
||||
|
||||
class VirtualTreeCtrlTest_OneRoot(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
self.tree = VirtualTreeCtrl(self.frame)
|
||||
self.tree.SetChildrenCount((), 1)
|
||||
self.tree.RefreshItems()
|
||||
|
||||
def testOneRoot(self):
|
||||
self.assertEqual(1, self.tree.GetCount())
|
||||
|
||||
def testDeleteRootItem(self):
|
||||
self.tree.SetChildrenCount((), 0)
|
||||
self.tree.RefreshItems()
|
||||
self.assertEqual(0, self.tree.GetCount())
|
||||
|
||||
def testAddOneChild(self):
|
||||
self.tree.SetChildrenCount((0,), 1)
|
||||
self.tree.RefreshItems()
|
||||
self.tree.ExpandAll()
|
||||
self.assertEqual(2, self.tree.GetCount())
|
||||
|
||||
def testAddTwoChildren(self):
|
||||
self.tree.SetChildrenCount((0,), 2)
|
||||
self.tree.RefreshItems()
|
||||
self.tree.ExpandAll()
|
||||
self.assertEqual(3, self.tree.GetCount())
|
||||
|
||||
|
||||
# TreeAPIHarmonizer tests
|
||||
|
||||
class TreeAPIHarmonizerTestCase(unittest.TestCase):
|
||||
style = wx.TR_DEFAULT_STYLE
|
||||
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
class HarmonizedTreeCtrl(treemixin.TreeAPIHarmonizer, self.TreeClass):
|
||||
pass
|
||||
self.tree = HarmonizedTreeCtrl(self.frame, style=self.style)
|
||||
self.populateTree()
|
||||
|
||||
def populateTree(self):
|
||||
self.root = self.tree.AddRoot('Root')
|
||||
self.item = self.tree.AppendItem(self.root, 'item')
|
||||
|
||||
|
||||
class TreeAPIHarmonizerCommonTests(object):
|
||||
def testSetItemImage(self):
|
||||
self.tree.SetItemImage(self.item, -1, wx.TreeItemIcon_Normal)
|
||||
self.assertEqual(-1,
|
||||
self.tree.GetItemImage(self.item, wx.TreeItemIcon_Normal))
|
||||
|
||||
def testGetColumnCount(self):
|
||||
self.assertEqual(0, self.tree.GetColumnCount())
|
||||
|
||||
def testGetItemType(self):
|
||||
self.assertEqual(0, self.tree.GetItemType(self.item))
|
||||
|
||||
def testGetMainWindow(self):
|
||||
self.assertEqual(self.tree, self.tree.GetMainWindow())
|
||||
|
||||
|
||||
class TreeAPIHarmonizerSingleSelectionTests(object):
|
||||
def testUnselectAll(self):
|
||||
self.tree.SelectItem(self.item)
|
||||
self.tree.UnselectAll()
|
||||
self.assertEqual([], self.tree.GetSelections())
|
||||
|
||||
def testGetSelections_NoSelection(self):
|
||||
self.tree.UnselectAll()
|
||||
self.assertEqual([], self.tree.GetSelections())
|
||||
|
||||
def testGetSelections_OneSelectedItem(self):
|
||||
self.tree.UnselectAll()
|
||||
self.tree.SelectItem(self.item)
|
||||
self.assertEqual([self.item], self.tree.GetSelections())
|
||||
|
||||
|
||||
class TreeAPIHarmonizerMultipleSelectionTests(object):
|
||||
style = wx.TR_DEFAULT_STYLE | wx.TR_MULTIPLE
|
||||
|
||||
def testUnselectAll(self):
|
||||
self.tree.SelectItem(self.item)
|
||||
self.tree.UnselectAll()
|
||||
self.assertEqual([], self.tree.GetSelections())
|
||||
|
||||
def testGetSelections_NoSelection(self):
|
||||
self.tree.UnselectAll()
|
||||
self.assertEqual([], self.tree.GetSelections())
|
||||
|
||||
def testGetSelections_OneSelectedItem(self):
|
||||
self.tree.UnselectAll()
|
||||
self.tree.SelectItem(self.item)
|
||||
self.assertEqual([self.item], self.tree.GetSelections())
|
||||
|
||||
def testGetSelections_TwoSelectedItems(self):
|
||||
item2 = self.tree.AppendItem(self.root, 'item 2')
|
||||
self.tree.ExpandAll()
|
||||
self.tree.UnselectAll()
|
||||
self.tree.SelectItem(item2)
|
||||
self.tree.SelectItem(self.item)
|
||||
self.assertEqual([self.item, item2], self.tree.GetSelections())
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeCtrlTestCase(TreeAPIHarmonizerTestCase):
|
||||
TreeClass = wx.TreeCtrl
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeCtrl_SingleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerSingleSelectionTests,
|
||||
TreeAPIHarmonizerWithTreeCtrlTestCase):
|
||||
pass
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeCtrl_MultipleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerMultipleSelectionTests,
|
||||
TreeAPIHarmonizerWithTreeCtrlTestCase):
|
||||
pass
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeListCtrlTestCase(TreeAPIHarmonizerTestCase):
|
||||
TreeClass = wx.gizmos.TreeListCtrl
|
||||
|
||||
def populateTree(self):
|
||||
self.tree.AddColumn('Column')
|
||||
super(TreeAPIHarmonizerWithTreeListCtrlTestCase, self).populateTree()
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeListCtrl_SingleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerSingleSelectionTests,
|
||||
TreeAPIHarmonizerWithTreeListCtrlTestCase):
|
||||
|
||||
def testGetColumnCount(self):
|
||||
self.assertEqual(1, self.tree.GetColumnCount())
|
||||
|
||||
def testGetMainWindow(self):
|
||||
self.assertNotEqual(self.tree, self.tree.GetMainWindow())
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithTreeListCtrl_MultipleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerMultipleSelectionTests,
|
||||
TreeAPIHarmonizerWithTreeListCtrlTestCase):
|
||||
|
||||
def testGetColumnCount(self):
|
||||
self.assertEqual(1, self.tree.GetColumnCount())
|
||||
|
||||
def testGetMainWindow(self):
|
||||
self.assertNotEqual(self.tree, self.tree.GetMainWindow())
|
||||
|
||||
def testGetSelections_TwoSelectedItems(self):
|
||||
''' Override TreeAPIHarmonizerMultipleSelectionTests.-
|
||||
testGetSelections_TwoSelectedItems, because
|
||||
TreeListCtrl.SelectItem needs an extra parameter. '''
|
||||
self.tree.UnselectAll()
|
||||
item2 = self.tree.AppendItem(self.root, 'item 2')
|
||||
self.tree.SelectItem(self.item)
|
||||
self.tree.SelectItem(item2, unselect_others=False)
|
||||
self.assertEqual([self.item, item2], self.tree.GetSelections())
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithCustomTreeCtrlTestCase(TreeAPIHarmonizerTestCase):
|
||||
TreeClass = wx.lib.customtreectrl.CustomTreeCtrl
|
||||
|
||||
|
||||
class TreeAPIHarmonizerCustomTreeCtrlTests(object):
|
||||
def testGetCheckItemType(self):
|
||||
item = self.tree.AppendItem(self.root, 'item', ct_type=1)
|
||||
self.assertEqual(1, self.tree.GetItemType(item))
|
||||
|
||||
def testGetRadioItemType(self):
|
||||
item = self.tree.AppendItem(self.root, 'item', ct_type=2)
|
||||
self.assertEqual(2, self.tree.GetItemType(item))
|
||||
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithCustomTreeCtrl_SingleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerSingleSelectionTests,
|
||||
TreeAPIHarmonizerCustomTreeCtrlTests,
|
||||
TreeAPIHarmonizerWithCustomTreeCtrlTestCase):
|
||||
pass
|
||||
|
||||
|
||||
class TreeAPIHarmonizerWithCustomTreeCtrl_MultipleSelection( \
|
||||
TreeAPIHarmonizerCommonTests, TreeAPIHarmonizerMultipleSelectionTests,
|
||||
TreeAPIHarmonizerCustomTreeCtrlTests,
|
||||
TreeAPIHarmonizerWithCustomTreeCtrlTestCase):
|
||||
|
||||
pass
|
||||
|
||||
|
||||
# ExpansionState tests
|
||||
|
||||
class ExpansionStateTreeCtrl(treemixin.ExpansionState, wx.TreeCtrl):
|
||||
pass
|
||||
|
||||
|
||||
class GetExpansionStateTestCase(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
self.tree = ExpansionStateTreeCtrl(self.frame)
|
||||
|
||||
def testEmptyTree(self):
|
||||
self.assertEqual([], self.tree.GetExpansionState())
|
||||
|
||||
def testRoot(self):
|
||||
self.tree.AddRoot('Root item')
|
||||
self.assertEqual([], self.tree.GetExpansionState())
|
||||
|
||||
def testRoot_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
self.assertEqual([], self.tree.GetExpansionState())
|
||||
|
||||
def testExpandedRoot_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
self.tree.Expand(root)
|
||||
self.assertEqual([()], self.tree.GetExpansionState())
|
||||
|
||||
def testExpandedRoot_Child_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.Expand(root)
|
||||
# Property should work too:
|
||||
self.assertEqual([()], self.tree.ExpansionState)
|
||||
|
||||
def testRoot_ExpandedChild_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.Expand(child)
|
||||
self.assertEqual([], self.tree.GetExpansionState())
|
||||
|
||||
def testExpandedRoot_ExpandedChild_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.Expand(child)
|
||||
self.tree.Expand(root)
|
||||
self.assertEqual([(), (0,)], self.tree.GetExpansionState())
|
||||
|
||||
def testExpandedRoot_ExpandedChild_ExpandedChild(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
grandGrandChild = self.tree.AppendItem(grandChild, 'Grandgrandchild')
|
||||
self.tree.Expand(root)
|
||||
self.tree.Expand(child)
|
||||
self.tree.Expand(grandChild)
|
||||
self.assertEqual([(), (0,), (0,0)], self.tree.GetExpansionState())
|
||||
|
||||
|
||||
class SetExpansionStateTestCase(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.frame = wx.Frame(None)
|
||||
self.tree = ExpansionStateTreeCtrl(self.frame)
|
||||
|
||||
def testEmptyTree(self):
|
||||
self.tree.SetExpansionState([])
|
||||
|
||||
def testRoot(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
self.tree.SetExpansionState([])
|
||||
self.failIf(self.tree.IsExpanded(root))
|
||||
|
||||
def testRoot_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
self.tree.SetExpansionState([])
|
||||
self.failIf(self.tree.IsExpanded(root))
|
||||
|
||||
def testExpandedRoot_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
self.tree.ExpansionState = [()] # Property should work too
|
||||
self.failUnless(self.tree.IsExpanded(root))
|
||||
|
||||
def testExpandedRoot_Child_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.SetExpansionState([()])
|
||||
self.failIf(self.tree.IsExpanded(child))
|
||||
|
||||
def testExpandedRoot_ExpandedChild_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.SetExpansionState([(), (0,)])
|
||||
self.failUnless(self.tree.IsExpanded(child))
|
||||
|
||||
def testRoot_ExpandedChild_Child(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
self.tree.SetExpansionState([(0,)])
|
||||
self.failIf(self.tree.IsExpanded(child))
|
||||
|
||||
def testExpandedRoot_ExpandedChild_ExpandedChild(self):
|
||||
root = self.tree.AddRoot('Root item')
|
||||
child = self.tree.AppendItem(root, 'Child')
|
||||
grandChild = self.tree.AppendItem(child, 'Grandchild')
|
||||
grandGrandChild = self.tree.AppendItem(grandChild, 'Grandgrandchild')
|
||||
self.tree.SetExpansionState([(), (0,), (0,0)])
|
||||
self.failUnless(self.tree.IsExpanded(grandChild))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app = wx.App(False)
|
||||
unittest.main()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user