def test_add_correct_sheet(): wb = Workbook() new_sheet = wb.create_sheet(0) wb.add_sheet(new_sheet) eq_(new_sheet, wb.worksheets[2])
def test_add_incorrect_sheet(): wb = Workbook() with pytest.raises(TypeError): wb.add_sheet("Test")
def test_add_invalid_worksheet_class_instance(): wb = Workbook() ws = AlternativeWorksheet(parent_workbook=wb) with pytest.raises(TypeError): wb.add_sheet(worksheet=ws)