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()