def test_collapse_empty(self):
""" Test for collapse function when the collapse origin is an empty
tabwidget. It's sibling can have an arbitrary layout and the result
would be such that this layout is transferred to the parent.
"""
# setup
root = EditorAreaWidget(editor_area=SplitEditorAreaPane(), parent=None)
tabwidget = root.tabwidget()
tabwidget.setParent(None)
left, left_left, left_right = self._setUp_collapse(parent=root)
right = EditorAreaWidget(editor_area=root.editor_area, parent=root)
root.leftchild = left
root.rightchild = right
# perform collapse on leftchild
right.collapse()
# test
# is the layout of root now same as left?
self.assertEqual(root.count(), 2)
self.assertEqual(root.leftchild, left_left)
self.assertEqual(root.rightchild, left_right)
# are the contents of left_left and left_right preserved
self.assertEqual(root.leftchild.tabwidget().count(), 2)
self.assertEqual(root.rightchild.tabwidget().count(), 2)
self.assertEqual(root.leftchild.tabwidget().currentIndex(), 1)
self.assertEqual(root.rightchild.tabwidget().currentIndex(), 0)
# what is the current active_tabwidget?
self.assertEqual(root.editor_area.active_tabwidget,
root.leftchild.tabwidget())
def _setUp_split(self, parent=None):
""" Sets up the root splitter for splitting. Returns this root.
parent : parent of the returned root
"""
root = EditorAreaWidget(editor_area=SplitEditorAreaPane(),
parent=parent)
btn0 = QtGui.QPushButton('0')
btn1 = QtGui.QPushButton('1')
tabwidget = root.tabwidget()
tabwidget.addTab(btn0, '0')
tabwidget.addTab(btn1, '1')
tabwidget.setCurrentIndex(1)
return root
def _setUp_collapse(self, parent=None):
""" Creates a root, its leftchild and rightchild, so that collapse can
be tested on one of the children.
Returns the root, leftchild and rightchild of such layout.
parent : parent of the returned root
"""
# setup leftchild
left = EditorAreaWidget(editor_area=SplitEditorAreaPane(), parent=None)
btn0 = QtGui.QPushButton('btn0')
btn1 = QtGui.QPushButton('btn1')
tabwidget = left.tabwidget()
tabwidget.addTab(btn0, '0')
tabwidget.addTab(btn1, '1')
tabwidget.setCurrentIndex(1)
# setup rightchild
right = EditorAreaWidget(editor_area=left.editor_area, parent=None)
btn2 = QtGui.QPushButton('btn2')
btn3 = QtGui.QPushButton('btn3')
tabwidget = right.tabwidget()
tabwidget.addTab(btn2, '2')
tabwidget.addTab(btn3, '3')
tabwidget.setCurrentIndex(0)
# setup root
root = EditorAreaWidget(editor_area=left.editor_area, parent=parent)
tabwidget = root.tabwidget()
tabwidget.setParent(None)
root.addWidget(left)
root.addWidget(right)
root.leftchild = left
root.rightchild = right
return root, left, right
def test_persistence(self):
""" Tests whether get_layout/set_layout work correctly by setting a
given layout and getting back the obtained layout.
"""
# setup the test layout - one horizontal split and one vertical split
# on the rightchild of horizontal split, where the top tabwidget of
# the vertical split is empty.
layout = Splitter(
Tabbed(PaneItem(id=0, width=600, height=600),
active_tab=0),
Splitter(Tabbed(PaneItem(id=-1, width=600, height=300),
active_tab=0),
Tabbed(PaneItem(id=1, width=600, height=300),
PaneItem(id=2, width=600, height=300),
active_tab=0),
orientation='vertical'),
orientation='horizontal')
# a total of 3 files are needed to give this layout - one on the
# leftchild of horizontal split, and the other two on the bottom
# tabwidget of the rightchild's vertical split
file0 = open(os.path.join(tempfile.gettempdir(), 'file0'), 'w+b')
file1 = open(os.path.join(tempfile.gettempdir(), 'file1'), 'w+b')
file2 = open(os.path.join(tempfile.gettempdir(), 'file2'), 'w+b')
# adding the editors
editor_area = SplitEditorAreaPane()
editor_area.create(parent=None)
editor_area.add_editor(Editor(obj=file0))
editor_area.add_editor(Editor(obj=file1))
editor_area.add_editor(Editor(obj=file2))
######## test set_layout #############
# set the layout
editor_area.set_layout(layout)
# file0 goes to left pane?
left = editor_area.control.leftchild
editor = editor_area._get_editor(left.tabwidget().widget(0))
self.assertEquals(editor.obj, file0)
# right pane is a splitter made of two panes?
right = editor_area.control.rightchild
self.assertFalse(right.is_leaf())
# right pane is vertical splitter?
self.assertEquals(right.orientation(), QtCore.Qt.Vertical)
# top pane of this vertical split is empty?
right_top = right.leftchild
self.assertTrue(right_top.is_empty())
# bottom pane is not empty?
right_bottom = right.rightchild
self.assertFalse(right_bottom.is_empty())
# file1 goes first on bottom pane?
editor = editor_area._get_editor(right_bottom.tabwidget().widget(0))
self.assertEquals(editor.obj, file1)
# file2 goes second on bottom pane?
editor = editor_area._get_editor(right_bottom.tabwidget().widget(1))
self.assertEquals(editor.obj, file2)
# file1 tab is active?
self.assertEquals(right_bottom.tabwidget().currentIndex(), 0)
######### test get_layout #############
# obtain layout
layout_new = editor_area.get_layout()
# is the top level a horizontal splitter?
self.assertIsInstance(layout_new, Splitter)
self.assertEquals(layout_new.orientation, 'horizontal')
# tests on left child
left = layout_new.items[0]
self.assertIsInstance(left, Tabbed)
self.assertEquals(left.items[0].id, 0)
# tests on right child
right = layout_new.items[1]
self.assertIsInstance(right, Splitter)
self.assertEquals(right.orientation, 'vertical')
# tests on top pane of right child
right_top = right.items[0]
self.assertIsInstance(right_top, Tabbed)
self.assertEquals(right_top.items[0].id, -1)
# tests on bottom pane of right child
right_bottom = right.items[1]
self.assertIsInstance(right_bottom, Tabbed)
self.assertEquals(right_bottom.items[0].id, 1)
self.assertEquals(right_bottom.items[1].id, 2)
def test_persistence(self):
""" Tests whether get_layout/set_layout work correctly by setting a
given layout and getting back the obtained layout.
"""
# setup the test layout - one horizontal split and one vertical split
# on the rightchild of horizontal split, where the top tabwidget of
# the vertical split is empty.
layout = Splitter(Tabbed(PaneItem(id=0, width=600, height=600),
active_tab=0),
Splitter(Tabbed(PaneItem(id=-1,
width=600,
height=300),
active_tab=0),
Tabbed(PaneItem(id=1, width=600,
height=300),
PaneItem(id=2, width=600,
height=300),
active_tab=0),
orientation='vertical'),
orientation='horizontal')
# a total of 3 files are needed to give this layout - one on the
# leftchild of horizontal split, and the other two on the bottom
# tabwidget of the rightchild's vertical split
file0 = open(os.path.join(tempfile.gettempdir(), 'file0'), 'w+b')
file1 = open(os.path.join(tempfile.gettempdir(), 'file1'), 'w+b')
file2 = open(os.path.join(tempfile.gettempdir(), 'file2'), 'w+b')
# adding the editors
editor_area = SplitEditorAreaPane()
editor_area.create(parent=None)
editor_area.add_editor(Editor(obj=file0, tooltip="test_tooltip0"))
editor_area.add_editor(Editor(obj=file1, tooltip="test_tooltip1"))
editor_area.add_editor(Editor(obj=file2, tooltip="test_tooltip2"))
######## test tooltips #############
self.assertEqual(editor_area.active_tabwidget.tabToolTip(0),
"test_tooltip0")
self.assertEqual(editor_area.active_tabwidget.tabToolTip(1),
"test_tooltip1")
self.assertEqual(editor_area.active_tabwidget.tabToolTip(2),
"test_tooltip2")
######## test set_layout #############
# set the layout
editor_area.set_layout(layout)
# file0 goes to left pane?
left = editor_area.control.leftchild
editor = editor_area._get_editor(left.tabwidget().widget(0))
self.assertEqual(editor.obj, file0)
# right pane is a splitter made of two panes?
right = editor_area.control.rightchild
self.assertFalse(right.is_leaf())
# right pane is vertical splitter?
self.assertEqual(right.orientation(), QtCore.Qt.Vertical)
# top pane of this vertical split is empty?
right_top = right.leftchild
self.assertTrue(right_top.is_empty())
# bottom pane is not empty?
right_bottom = right.rightchild
self.assertFalse(right_bottom.is_empty())
# file1 goes first on bottom pane?
editor = editor_area._get_editor(right_bottom.tabwidget().widget(0))
self.assertEqual(editor.obj, file1)
# file2 goes second on bottom pane?
editor = editor_area._get_editor(right_bottom.tabwidget().widget(1))
self.assertEqual(editor.obj, file2)
# file1 tab is active?
self.assertEqual(right_bottom.tabwidget().currentIndex(), 0)
######### test get_layout #############
# obtain layout
layout_new = editor_area.get_layout()
# is the top level a horizontal splitter?
self.assertIsInstance(layout_new, Splitter)
self.assertEqual(layout_new.orientation, 'horizontal')
# tests on left child
left = layout_new.items[0]
self.assertIsInstance(left, Tabbed)
self.assertEqual(left.items[0].id, 0)
# tests on right child
right = layout_new.items[1]
self.assertIsInstance(right, Splitter)
self.assertEqual(right.orientation, 'vertical')
# tests on top pane of right child
right_top = right.items[0]
self.assertIsInstance(right_top, Tabbed)
self.assertEqual(right_top.items[0].id, -1)
# tests on bottom pane of right child
right_bottom = right.items[1]
self.assertIsInstance(right_bottom, Tabbed)
self.assertEqual(right_bottom.items[0].id, 1)
self.assertEqual(right_bottom.items[1].id, 2)