def test_create_workbook_default_sheet(fmt): library = Files() library.create_workbook(fmt=fmt) assert library.list_worksheets() == ["Sheet"] library.create_worksheet("Test") assert library.list_worksheets() == ["Sheet", "Test"]
def generate_data(user_count=100, feedback_count=5): fake = Faker() files = Files() files.create_workbook("userdata.xlsx") files.create_worksheet("profile", exist_ok=True) files.create_worksheet("feedback", exist_ok=True) users = [] names = [] for i in range(user_count): userprofile = fake.profile() if userprofile["name"] in names: print("name %s already exists", userprofile["name"]) continue userprofile.pop("website") userprofile.pop("current_location") files.append_rows_to_worksheet(userprofile, "profile", header=True) users.append(userprofile) for i in range(1, 16): selected_user = users[i] feedback = { "name": selected_user["name"], "feedback": choice(good_feedback) } print(feedback) files.append_rows_to_worksheet(feedback, "feedback", header=True) for i in range(17, 22): selected_user = users[i] feedback = { "name": selected_user["name"], "feedback": choice(neutral_feedback) } print(feedback) files.append_rows_to_worksheet(feedback, "feedback", header=True) for i in range(23, 28): selected_user = users[i] feedback = { "name": selected_user["name"], "feedback": choice(bad_feedback) } print(feedback) files.append_rows_to_worksheet(feedback, "feedback", header=True) files.remove_worksheet("Sheet") files.save_workbook()