Test selection containers more comprehensively.

Author: jasongrout

python/ipywidgets/ipywidgets/widgets/tests/test_selectioncontainer.py

@@ -5,55 +5,164 @@
 
 from traitlets import TraitError
 
-from ipywidgets.widgets import Accordion, Tab, Stack, HTML
+from ipywidgets.widgets import Accordion, Tab, Stacked, HTML
 
+class TestTab(TestCase):
+
+    def setUp(self):
+        self.children = [HTML('0'), HTML('1')]
+        self.widget = Tab
+
+    def test_selected_index_none(self):
+        widget = self.widget(self.children, selected_index=None)
+        state = widget.get_state()
+        assert state['selected_index'] == 0
+
+    def test_selected_index_default(self):
+        widget = self.widget(self.children)
+        state = widget.get_state()
+        assert state['selected_index'] == 0
+
+    def test_selected_index(self):
+        widget = self.widget(self.children, selected_index=1)
+        state = widget.get_state()
+        assert state['selected_index'] == 1
+
+    def test_selected_index_out_of_bounds(self):
+        with self.assertRaises(TraitError):
+            self.widget(self.children, selected_index=-1)
+
+    def test_children_position_argument(self):
+        self.widget(self.children)
+
+    def test_titles(self):
+        widget = self.widget(self.children, selected_index=None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles == ('', '')
+
+        widget.set_title(1, 'Title 1')
+        assert widget.get_state()['titles'] == ('', 'Title 1')
+        assert widget.titles[1] == 'Title 1'
+        assert widget.get_title(1) == 'Title 1'
+
+        # Backwards compatible with 7.x api
+        widget.set_title(1, None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles[1] == ''
+        assert widget.get_title(1) == ''
+
+        with self.assertRaises(IndexError):
+            widget.set_title(2, 'out of bounds')
+        with self.assertRaises(IndexError):
+            widget.get_title(2)
+
+        widget.children = tuple(widget.children[:1])
+        assert len(widget.children) == 1
+        assert widget.titles == ('',)
 
 class TestAccordion(TestCase):
 
     def setUp(self):
         self.children = [HTML('0'), HTML('1')]
+        self.widget = Accordion
 
     def test_selected_index_none(self):
-        accordion = Accordion(self.children, selected_index=None)
-        state = accordion.get_state()
+        widget = self.widget(self.children, selected_index=None)
+        state = widget.get_state()
+        assert state['selected_index'] is None
+
+    def test_selected_index_default(self):
+        widget = self.widget(self.children)
+        state = widget.get_state()
         assert state['selected_index'] is None
 
     def test_selected_index(self):
-        accordion = Accordion(self.children, selected_index=1)
-        state = accordion.get_state()
+        widget = self.widget(self.children, selected_index=1)
+        state = widget.get_state()
         assert state['selected_index'] == 1
 
     def test_selected_index_out_of_bounds(self):
         with self.assertRaises(TraitError):
-            Accordion(self.children, selected_index=-1)
+            self.widget(self.children, selected_index=-1)
 
     def test_children_position_argument(self):
-        Accordion(self.children)
-        Tab(self.children)
-        Stack(self.children)
+        self.widget(self.children)
 
+    def test_titles(self):
+        widget = self.widget(self.children, selected_index=None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles == ('', '')
+
+        widget.set_title(1, 'Title 1')
+        assert widget.get_state()['titles'] == ('', 'Title 1')
+        assert widget.titles[1] == 'Title 1'
+        assert widget.get_title(1) == 'Title 1'
+
+        # Backwards compatible with 7.x api
+        widget.set_title(1, None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles[1] == ''
+        assert widget.get_title(1) == ''
+
+        with self.assertRaises(IndexError):
+            widget.set_title(2, 'out of bounds')
+        with self.assertRaises(IndexError):
+            widget.get_title(2)
+
+        widget.children = tuple(widget.children[:1])
+        assert len(widget.children) == 1
+        assert widget.titles == ('',)
+
+class TestStacked(TestCase):
+
+    def setUp(self):
+        self.children = [HTML('0'), HTML('1')]
+        self.widget = Stacked
+
+    def test_selected_index_none(self):
+        widget = self.widget(self.children, selected_index=None)
+        state = widget.get_state()
+        assert state['selected_index'] is None
+
+    def test_selected_index_default(self):
+        widget = self.widget(self.children)
+        state = widget.get_state()
+        assert state['selected_index'] is None
+
+    def test_selected_index(self):
+        widget = self.widget(self.children, selected_index=1)
+        state = widget.get_state()
+        assert state['selected_index'] == 1
+
+    def test_selected_index_out_of_bounds(self):
+        with self.assertRaises(TraitError):
+            self.widget(self.children, selected_index=-1)
+
+    def test_children_position_argument(self):
+        self.widget(self.children)
 
     def test_titles(self):
-        accordion = Accordion(self.children, selected_index=None)
-        assert accordion.get_state()['titles'] == ('', '')
-        assert accordion.titles == ('', '')
+        widget = self.widget(self.children, selected_index=None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles == ('', '')
 
-        accordion.set_title(1, 'Title 1')
-        assert accordion.get_state()['titles'] == ('', 'Title 1')
-        assert accordion.titles[1] == 'Title 1'
-        assert accordion.get_title(1) == 'Title 1'
+        widget.set_title(1, 'Title 1')
+        assert widget.get_state()['titles'] == ('', 'Title 1')
+        assert widget.titles[1] == 'Title 1'
+        assert widget.get_title(1) == 'Title 1'
 
         # Backwards compatible with 7.x api
-        accordion.set_title(1, None)
-        assert accordion.get_state()['titles'] == ('', '')
-        assert accordion.titles[1] == ''
-        assert accordion.get_title(1) == ''
+        widget.set_title(1, None)
+        assert widget.get_state()['titles'] == ('', '')
+        assert widget.titles[1] == ''
+        assert widget.get_title(1) == ''
 
         with self.assertRaises(IndexError):
-            accordion.set_title(2, 'out of bounds')
+            widget.set_title(2, 'out of bounds')
         with self.assertRaises(IndexError):
-            accordion.get_title(2)
+            widget.get_title(2)
+
+        widget.children = tuple(widget.children[:1])
+        assert len(widget.children) == 1
+        assert widget.titles == ('',)
 
-        accordion.children = tuple(accordion.children[:1])
-        assert len(accordion.children) == 1
-        assert accordion.titles == ('',)