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)