def test_WorkbookElement_getSheetByIndex_var_index(self):
_test_el = WorkbookElement()
self._remove_patched_function('_getSheets')
with patch.object(_test_el, 'getSheets') as _getSheets:
_getSheets.return_value = ['sheet1', 'sheet2', 'sheet3']
_sheets1 = _test_el.getSheetByIndex(None)
self.assertIsNone(_sheets1)
_sheets2 = _test_el.getSheetByIndex(-1)
self.assertIsNone(_sheets2)
_sheets3 = _test_el.getSheetByIndex(3)
self.assertIsNone(_sheets3)
_sheets4 = _test_el.getSheetByIndex(4)
self.assertIsNone(_sheets4)
self.assertEqual(_getSheets.call_count, 4)
self._assert_init_methods_called()
def test_WorkbookElement_getSheetByIndex_index_within_len_sheets(self):
_test_el = WorkbookElement()
self._remove_patched_function('_getSheets')
with patch.object(_test_el, 'getSheets') as _getSheets:
_getSheets.return_value = ['sheet1', 'sheet2', 'sheet3']
_sheets = _test_el.getSheetByIndex(2)
_getSheets.assert_called_once_with()
self.assertEqual(_sheets, 'sheet3')
self._assert_init_methods_called()